Skip to content

NNModelStandard tests#75

Merged
sebapersson merged 4 commits into
mainfrom
nn_standard_tests
May 19, 2026
Merged

NNModelStandard tests#75
sebapersson merged 4 commits into
mainfrom
nn_standard_tests

Conversation

@sebapersson
Copy link
Copy Markdown
Collaborator

@sebapersson sebapersson commented May 19, 2026

Add tests for NN yaml export via testing the roundtrip. This PR also removes some old code from the petab_sciml/standard directory.

See also #11

@sebapersson sebapersson requested review from BSnelling and dilpath May 19, 2026 12:35
Copy link
Copy Markdown
Member

@dilpath dilpath left a comment

Choose a reason for hiding this comment

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

Looks good. Currently missing is a test that the reconstructed pytorch module matches the original pytorch module, e.g. by evaluating the module forward method and seeing that the output is the same after reconstruction. But this is already nice, thanks!

Comment thread tests/test_nn_yaml_export.py Outdated
@sebapersson
Copy link
Copy Markdown
Collaborator Author

Currently missing is a test that the reconstructed pytorch module matches the original pytorch module, e.g. by evaluating the module forward method and seeing that the output is the same after reconstruction.

Fair point, I can likely quite easily add it to the PR

@sebapersson sebapersson merged commit 7f910b9 into main May 19, 2026
2 checks passed
@sebapersson sebapersson deleted the nn_standard_tests branch May 19, 2026 15:57
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.

3 participants