Description
Why are we defining a new string type? An alias is sufficient if we even need it, but I think str is just fine. Also all PyTensor variables have a name attribute we don't need to call getattr(variable, "name") it may just be None
|
VarName = NewType("VarName", str) |
|
def get_var_name(var) -> VarName: |
|
"""Get an appropriate, plain variable name for a variable.""" |
|
return VarName(str(getattr(var, "name", var))) |
