From b2a3d1e648fc56355b176d84eb21648a068055fa Mon Sep 17 00:00:00 2001 From: Will Dean Date: Tue, 28 Jan 2025 17:52:00 +0100 Subject: [PATCH] specify the mode --- pymc/pytensorf.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pymc/pytensorf.py b/pymc/pytensorf.py index dc0c157919..8dff15a45e 100644 --- a/pymc/pytensorf.py +++ b/pymc/pytensorf.py @@ -180,7 +180,8 @@ def extract_obs_data(x: TensorVariable) -> np.ndarray: from pymc.data import has_random_ancestor if not has_random_ancestor(x): - return x.eval() + cheap_eval_mode = Mode(linker="py", optimizer=None) + return x.eval(mode=cheap_eval_mode) raise TypeError(f"Data cannot be extracted from {x}")