Update utils.py

torchmd · Apr 19, 2024 · 433d59c · 433d59c
1 parent 66521a4
commit 433d59c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/utils.py b/tests/utils.py
@@ -10,7 +10,7 @@
 
 def load_example_args(model_name, remove_prior=False, config_file=None, **kwargs):
     if config_file is None:
-        if model_name == "tensornet" or model_name == "equivariant-tensornet":
+        if model_name == "tensornet":
             config_file = join(dirname(dirname(__file__)), "examples", "TensorNet-QM9.yaml")
         else:
             config_file = join(dirname(dirname(__file__)), "examples", "ET-QM9.yaml")