File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change @@ -160,6 +160,12 @@ def fgraph_from_model(
160
160
"Nested sub-models cannot be converted to fgraph. Convert the parent model instead"
161
161
)
162
162
163
+ check_for_gp_vars = [
164
+ k for x in ["_rotated_" , "_hsgp_coeffs_" ] for k in model .named_vars .keys () if x in k
165
+ ]
166
+ if len (check_for_gp_vars ) > 0 :
167
+ warnings .warn ("Unable to clone Gaussian Process Variables" , UserWarning )
168
+
163
169
# Collect PyTensor variables
164
170
rvs_to_values = model .rvs_to_values
165
171
rvs = list (rvs_to_values .keys ())
@@ -393,11 +399,6 @@ def clone_model(model: Model) -> Model:
393
399
z = pm.Deterministic("z", clone_x + 1)
394
400
395
401
"""
396
- check_for_gp_vars = [
397
- k for x in ["_rotated_" , "_hsgp_coeffs_" ] for k in model .named_vars .keys () if x in k
398
- ]
399
- if len (check_for_gp_vars ) > 0 :
400
- warnings .warn ("Unable to clone Gaussian Process Variables" , UserWarning )
401
402
return model_from_fgraph (fgraph_from_model (model )[0 ], mutate_fgraph = True )
402
403
403
404
You can’t perform that action at this time.
0 commit comments