Fix Sliceable instance check

tum-pbs · Nov 24, 2023 · a5014f8 · a5014f8
1 parent 1bf1dea
commit a5014f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/phiml/math/magic.py b/phiml/math/magic.py
@@ -78,7 +78,7 @@ class _SliceableType(type):
     def __instancecheck__(self, instance):
         if isinstance(instance, str):
             return False
-        return isinstance(instance, Shaped) and hasattr(instance, '__getitem__')
+        return isinstance(instance, Shaped) and (hasattr(instance, '__getitem__') or isinstance(instance, PhiTreeNode))
 
     def __subclasscheck__(self, subclass):
         if subclass == str: