diff --git a/mtenn/config.py b/mtenn/config.py index 605ba1e..5e377f5 100644 --- a/mtenn/config.py +++ b/mtenn/config.py @@ -648,7 +648,7 @@ def massage_irreps(cls, values): irreps = values["irreps_hidden"] # First see if this string should be converted into a dict if isinstance(irreps, str): - if ("," in irreps) and (":" in irreps): + if ":" in irreps: orig_irreps = irreps irreps = [i.split(":") for i in irreps.split(",")] try: