Skip to content

Commit f6cd837

Browse files
committed
Appease mypy
1 parent 26f9aaa commit f6cd837

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pymc/model/transform/basic.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -72,13 +72,13 @@ def remove_minibatched_nodes(model: Model) -> Model:
7272
for inp, out in zip(var.inputs, var.outputs):
7373
replacements[out] = inp
7474

75-
old_outs, old_coords, old_dim_lengths = fgraph.outputs, fgraph._coords, fgraph._dim_lengths
75+
old_outs, old_coords, old_dim_lengths = fgraph.outputs, fgraph._coords, fgraph._dim_lengths # type: ignore[attr-defined]
7676
# Using `rebuild_strict=False` means all coords, names, and dim information is lost
7777
# So we need to restore it from the old fgraph
78-
new_outs = clone_replace(old_outs, replacements, rebuild_strict=False)
78+
new_outs = clone_replace(old_outs, replacements, rebuild_strict=False) # type: ignore[arg-type]
7979
for old_out, new_out in zip(old_outs, new_outs):
8080
new_out.name = old_out.name
8181
fgraph = FunctionGraph(outputs=new_outs, clone=False)
82-
fgraph._coords = old_coords
83-
fgraph._dim_lengths = old_dim_lengths
82+
fgraph._coords = old_coords # type: ignore[attr-defined]
83+
fgraph._dim_lengths = old_dim_lengths # type: ignore[attr-defined]
8484
return model_from_fgraph(fgraph, mutate_fgraph=True)

0 commit comments

Comments
 (0)