We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent eced55e commit cbf439aCopy full SHA for cbf439a
tests/models/test_sevennet.py
@@ -34,8 +34,7 @@ def pretrained_sevenn_model():
34
"""Load a pretrained SevenNet model for testing."""
35
cp = sevenn.util.load_checkpoint(model_name)
36
37
- backend = "e3nn"
38
- model_loaded = cp.build_model(backend)
+ model_loaded = cp.build_model()
39
model_loaded.set_is_batch_data(True)
40
41
return model_loaded.to(DEVICE)
0 commit comments