Skip to content

Add abstract base class for Model and Trainer - #598

Merged
Luthaf merged 1 commit into
mainfrom
model-trainer-base-classes
May 27, 2025
Merged

Add abstract base class for Model and Trainer#598
Luthaf merged 1 commit into
mainfrom
model-trainer-base-classes

Conversation

@Luthaf

@Luthaf Luthaf commented May 21, 2025

Copy link
Copy Markdown
Member

These allow to both document and enforce what function a model and trainer must have to be compatible with metatrain

Contributor (creator of pull-request) checklist

  • Tests updated (for new features and bugfixes)?
  • Documentation updated (for new features)?
  • Issue referenced (for PRs that solve an issue)?

Reviewer checklist

  • CHANGELOG updated with public API or any other important changes?

📚 Documentation preview 📚: https://metatrain--598.org.readthedocs.build/en/598/

@Luthaf
Luthaf requested a review from PicoCentauri May 21, 2025 17:32
@Luthaf
Luthaf force-pushed the model-trainer-base-classes branch from be52f98 to 0fbf85b Compare May 22, 2025 09:36

@PicoCentauri PicoCentauri left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks clean and is well reflecting our design. I think the base class will be useful for new architecture to be added!

Comment thread src/metatrain/utils/abc.py
Comment thread src/metatrain/utils/abc.py
@Luthaf
Luthaf requested a review from PicoCentauri May 27, 2025 13:18
These allow to both document and enforce what function
a model and trainer must have to be compatible with
metatrain
@Luthaf
Luthaf force-pushed the model-trainer-base-classes branch from 0fbf85b to 371fb4e Compare May 27, 2025 15:27
@Luthaf
Luthaf merged commit d3a33c1 into main May 27, 2025
@Luthaf
Luthaf deleted the model-trainer-base-classes branch May 27, 2025 15:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants