-
Notifications
You must be signed in to change notification settings - Fork 1
Train code split, Part 4: moving some of the model.py code into dataset.py #40
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Overall, I think this is a good start. Some of the documentation needs some additional work and details and there are some small code improvements that I think are worthwhile in this PR. Certainly a lot of work for us in improving the code readability over time as well 😂
@emersodb I have addressed all of your comments, please check again :) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Overall, looks good to go. Just a a couple small final comments.
04d817f Trainer: Changing all literals to enums (#48) 33f9306 Adding ignore for pip 25.2 vulnerability, removing stale ones (#50) c8d30ca Remove mkdocs build dir, add to gitignore (#49) c01121c End-to-end Evaluation Script Example (#45) d231a4a Add Nearest Neighbor Distance Ratio and Epsilon Identifiability Privacy Metrics (#42) 53e423b Adding Mean F1 Score Difference and Hitting Rate Metrics (#39) 91150fd Adding in Hellinger and pMSE metrics (#38) 746644e Tightening Ruff Configuration (#46) 580d55f Adding data_split_ratios to both the diffusion config and the classifier config (#47) 72863be Refactoring core.logger into common.logger and removing it (#41) bef53bf Train code split, Part 4: moving some of the model.py code into dataset.py (#40) 80b0154 Upgrading pip to latest version to solve security issue (#44) 83beba6 New mypy flow and fixes to typing issues that were discovered (#43) 7e77f37 Train code split, Part 3: moving some of the model.py code into sampler.py (#9) git-subtree-dir: deps/midst-toolkit git-subtree-split: 04d817f
PR Type
Fix
Short Description
Clickup Ticket(s): https://app.clickup.com/t/868ew1hg6
This is the last one of the code split tickets.
Moving data related code to
dataset.py
, adding docstrings and type hints where missing and doing minor refactorings (major ones coming up next).Moving
data_loaders.py
to theclavaddpm
folder and adding missing docstrings and type hints in that file.Also adding missing docstrings and type hints in
model.py
.Tests Added
NA