Fix typing in model.transform.basic

Author: michaelosthege

pymc/model/transform/basic.py

@@ -54,6 +54,8 @@ def prune_vars_detached_from_observed(model: Model) -> Model:
 
 
 def parse_vars(model: Model, vars: Union[ModelVariable, Sequence[ModelVariable]]) -> List[Variable]:
-    if not isinstance(vars, (list, tuple)):
-        vars = (vars,)
-    return [model[var] if isinstance(var, str) else var for var in vars]
+    if isinstance(vars, (list, tuple)):
+        vars_seq = vars
+    else:
+        vars_seq = (vars,)
+    return [model[var] if isinstance(var, str) else var for var in vars_seq]

scripts/run_mypy.py

@@ -41,7 +41,6 @@
 pymc/logprob/utils.py
 pymc/model/core.py
 pymc/model/fgraph.py
-pymc/model/transform/basic.py
 pymc/model/transform/conditioning.py
 pymc/printing.py
 pymc/pytensorf.py