shortened TestModelCopy by removing assigned variables before comparison

pymc-devs · Sep 30, 2024 · fb00f85 · fb00f85
1 parent 07106ec
commit fb00f85
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/tests/model/test_core.py b/tests/model/test_core.py
@@ -1782,12 +1782,10 @@ def test_copy_model(self, copy_method) -> None:
                 samples=1, random_seed=42
             )
 
-        simple_model_prior_predictive_val = simple_model_prior_predictive["prior"]["y"].values
-        copy_simple_model_prior_predictive_val = copy_simple_model_prior_predictive["prior"][
-            "y"
-        ].values
-
-        assert simple_model_prior_predictive_val == copy_simple_model_prior_predictive_val
+        assert (
+            simple_model_prior_predictive["prior"]["y"].values
+            == copy_simple_model_prior_predictive["prior"]["y"].values
+        )
 
         with copy_simple_model:
             z = pm.Deterministic("z", copy_simple_model["alpha"] + 1)