check for variable having inputvars

Author: williambdean

pymc/pytensorf.py

@@ -179,7 +179,7 @@ def extract_obs_data(x: TensorVariable) -> np.ndarray:
 
     from pymc.logprob.utils import rvs_in_graph
 
-    if not rvs_in_graph(x):
+    if not inputvars(x) and not rvs_in_graph(x):
         cheap_eval_mode = Mode(linker="py", optimizer=None)
         return x.eval(mode=cheap_eval_mode)
 

tests/test_pytensorf.py

@@ -207,6 +207,13 @@ def test_pytensor_operations(self):
         assert isinstance(res, np.ndarray)
         np.testing.assert_array_equal(res, np.array([4, 7, 10]))
 
+    def test_pytensor_operations_raises(self):
+        x = pt.scalar("x")
+        target = 1 + 3 * x
+
+        with pytest.raises(TypeError, match="Data cannot be extracted from"):
+            extract_obs_data(target)
+
 
 @pytest.mark.parametrize("input_dtype", ["int32", "int64", "float32", "float64"])
 def test_convert_data(input_dtype):