Fix TensorStack.native(str)

tum-pbs · Jan 27, 2025 · 40d3a2b · 40d3a2b
1 parent 0976cc8
commit 40d3a2b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/phiml/math/_tensors.py b/phiml/math/_tensors.py
@@ -2459,6 +2459,8 @@ def process_group(g):  # returns Shape or Ellipsis
             for name in g:
                 assert name in shape, f"When specifying a group by dim names, all dims must be present but {name} is not part of {shape}"
         return shape.only(g, reorder=True)
+    if isinstance(groups, str):
+        groups = groups.split(',')
     groups = [process_group(g) for g in groups]
     if Ellipsis in groups:
         ellipsis_dims = shape.without([g for g in groups if g is not Ellipsis])