From 433d59cda2e1511ba5be4513478aa27018f5435d Mon Sep 17 00:00:00 2001 From: Guillem Simeon <55756547+guillemsimeon@users.noreply.github.com> Date: Fri, 19 Apr 2024 10:58:06 +0200 Subject: [PATCH] Update utils.py --- tests/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/utils.py b/tests/utils.py index 0de2337c8..ef8bcddb9 100644 --- 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")