fix uts

iProzd · Feb 27, 2024 · 847565a · 847565a
1 parent c2f306c
commit 847565a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/source/tests/common/test_examples.py b/source/tests/common/test_examples.py
@@ -39,7 +39,6 @@
     p_examples / "zinc_protein" / "zinc_se_a_mask.json",
     p_examples / "dos" / "train" / "input.json",
     p_examples / "spin" / "se_e2_a" / "input_tf.json",
-    p_examples / "spin" / "se_e2_a" / "input_torch.json",
     p_examples / "dprc" / "normal" / "input.json",
     p_examples / "dprc" / "pairwise" / "input.json",
     p_examples / "dprc" / "generalized_force" / "input.json",

diff --git a/source/tests/consistent/descriptor/test_se_e2_a.py b/source/tests/consistent/descriptor/test_se_e2_a.py
@@ -143,6 +143,7 @@ def setUp(self):
             type_one_side,
             excluded_types,
             precision,
+            env_protection,
         ) = self.param
         if not type_one_side:
             idx = np.argsort(self.atype)