Skip to content

Commit cbf439a

Browse files
authored
fix test sevennet (#393)
1 parent eced55e commit cbf439a

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/models/test_sevennet.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,7 @@ def pretrained_sevenn_model():
3434
"""Load a pretrained SevenNet model for testing."""
3535
cp = sevenn.util.load_checkpoint(model_name)
3636

37-
backend = "e3nn"
38-
model_loaded = cp.build_model(backend)
37+
model_loaded = cp.build_model()
3938
model_loaded.set_is_batch_data(True)
4039

4140
return model_loaded.to(DEVICE)

0 commit comments

Comments
 (0)