Skip to content

Commit

Permalink
remove redundant visnet set_config test
Browse files Browse the repository at this point in the history
  • Loading branch information
fyng committed Feb 5, 2024
1 parent a406012 commit 5f161a3
Showing 1 changed file with 1 addition and 11 deletions.
12 changes: 1 addition & 11 deletions mtenn/tests/test_model_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,21 +113,11 @@ def test_random_seed_visnet_from_pyg():
)

rand_config = ViSNetModelConfig(model=model)
set_config = ViSNetModelConfig(rand_seed=10)

rand_model1 = rand_config.build()
rand_model2 = rand_config.build()
set_model1 = set_config.build()
set_model2 = set_config.build()

rand_equal = [
(p1 == p2).all()
for p1, p2 in zip(rand_model1.parameters(), rand_model2.parameters())
for p1, p2 in zip(rand_model1.parameters(), model.parameters())
]
assert sum(rand_equal) < len(rand_equal)

set_equal = [
(p1 == p2).all()
for p1, p2 in zip(set_model1.parameters(), set_model2.parameters())
]
assert sum(set_equal) == len(set_equal)

0 comments on commit 5f161a3

Please sign in to comment.