Update test_equivariance.py

guillemsimeon · web-flow · commit d47493a4c076 · 2024-04-19T11:05:40.000+02:00
fix
diff --git a/tests/test_equivariance.py b/tests/test_equivariance.py
@@ -38,15 +38,15 @@ def test_vector_equivariance(model_name):
             [-0.0626055, 0.3134752, 0.9475304],
         ]
     )
-    if model_name == "equivariant_transformer"
+    if model_name == "equivariant_transformer":
         model = create_model(
             load_example_args(
                 model_name,
                 prior_model=None,
                 output_model="VectorOutput",
             )
         )
-    if model_name == "tensornet"
+    if model_name == "tensornet":
         model = create_model(
             load_example_args(
                 model_name,

Original file line number	Diff line number	Diff line change
`@@ -38,15 +38,15 @@ def test_vector_equivariance(model_name):`
`38`	`38`	`[-0.0626055, 0.3134752, 0.9475304],`
`39`	`39`	`]`
`40`	`40`	`)`
`41`		`- if model_name == "equivariant_transformer"`
	`41`	`+ if model_name == "equivariant_transformer":`
`42`	`42`	`model = create_model(`
`43`	`43`	`load_example_args(`
`44`	`44`	`model_name,`
`45`	`45`	`prior_model=None,`
`46`	`46`	`output_model="VectorOutput",`
`47`	`47`	`)`
`48`	`48`	`)`
`49`		`- if model_name == "tensornet"`
	`49`	`+ if model_name == "tensornet":`
`50`	`50`	`model = create_model(`
`51`	`51`	`load_example_args(`
`52`	`52`	`model_name,`