Skip to content

Added tests#6

Merged
MatteoMazzonelli merged 1 commit into
mainfrom
test/model_output
Jul 1, 2025
Merged

Added tests#6
MatteoMazzonelli merged 1 commit into
mainfrom
test/model_output

Conversation

@MatteoMazzonelli
Copy link
Copy Markdown
Collaborator

  • Added test to check model output
  • Removed redundant pre-commit check (Flake8)

@MatteoMazzonelli MatteoMazzonelli self-assigned this Jul 1, 2025
@MatteoMazzonelli MatteoMazzonelli added the enhancement New feature or request label Jul 1, 2025
Comment thread tests/create_reference.py
torch.manual_seed(0)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

num_nodes = 1
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

why do you test only with num_nodes=1?

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

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

it's just a dummy input. It's to test that the models are always the same and did not break

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

yeah but then there is no message passing?

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

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

There is the self loop ;-)

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

ok then haha

@MatteoMazzonelli MatteoMazzonelli removed the request for review from celiacintas July 1, 2025 11:54
@MatteoMazzonelli MatteoMazzonelli merged commit 9399bdc into main Jul 1, 2025
4 checks passed
@MatteoMazzonelli MatteoMazzonelli deleted the test/model_output branch July 1, 2025 11:57
romeokienzler pushed a commit that referenced this pull request Apr 21, 2026
romeokienzler pushed a commit that referenced this pull request Apr 21, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants