Skip to content

Commit fb011bf

Browse files
committed
Align get_var_name with the current behavior
if var.name is None, current get_var_name returns "None", not str(var)
1 parent 5148aa3 commit fb011bf

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

pymc/util.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -214,12 +214,7 @@ def get_default_varnames(var_iterator, include_transformed):
214214

215215
def get_var_name(var) -> str:
216216
"""Get an appropriate, plain variable name for a variable."""
217-
if isinstance(var, str):
218-
return var
219-
elif var.name is not None:
220-
return str(var.name)
221-
else:
222-
return str(var)
217+
return var if isinstance(var, str) else str(var.name)
223218

224219

225220
def get_transformed(z):

0 commit comments

Comments
 (0)