Skip to content

Conversation

lotif
Copy link
Collaborator

@lotif lotif commented Sep 17, 2025

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 the clavaddpm folder and adding missing docstrings and type hints in that file.

Also adding missing docstrings and type hints in model.py.

Tests Added

NA

Base automatically changed from sampler-module to main September 19, 2025 15:39
Copy link
Collaborator

@emersodb emersodb left a 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 😂

@lotif
Copy link
Collaborator Author

lotif commented Sep 25, 2025

@emersodb I have addressed all of your comments, please check again :)

Copy link
Collaborator

@emersodb emersodb left a 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.

@lotif lotif merged commit bef53bf into main Sep 26, 2025
5 checks passed
@lotif lotif deleted the data-module branch September 26, 2025 16:13
bzamanlooy pushed a commit that referenced this pull request Oct 10, 2025
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
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.

7 participants