fix missing @

materialsproject · Oct 22, 2024 · 8d11d91 · 8d11d91
1 parent b2c97bc
commit 8d11d91
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/transformations/test_advanced_transformations.py b/tests/transformations/test_advanced_transformations.py
@@ -182,8 +182,7 @@ def test_apply_transformation(self):
         for struct_trafo in alls:
             assert "energy" not in struct_trafo
 
-    pytest.skip(reason="dgl don't support torch 2.4.1+, #4073")
-
+    @pytest.mark.skip(reason="dgl don't support torch 2.4.1+, #4073")
     def test_m3gnet(self):
         pytest.importorskip("matgl")
         enum_trans = EnumerateStructureTransformation(refine_structure=True, sort_criteria="m3gnet_relax")
@@ -199,7 +198,7 @@ def test_m3gnet(self):
         # Check ordering of energy/atom
         assert alls[0]["energy"] / alls[0]["num_sites"] <= alls[-1]["energy"] / alls[-1]["num_sites"]
 
-    @pytest.skip(reason="dgl don't support torch 2.4.1+, #4073")
+    @pytest.mark.skip(reason="dgl don't support torch 2.4.1+, #4073")
     def test_callable_sort_criteria(self):
         matgl = pytest.importorskip("matgl")
         from matgl.ext.ase import Relaxer