[field] Fix Field.shape

tum-pbs · Aug 10, 2024 · bfeaa1d · bfeaa1d
1 parent f25f90f
commit bfeaa1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/phi/field/_field.py b/phi/field/_field.py
@@ -207,7 +207,7 @@ def shape(self) -> Shape:
         """
         if self.is_staggered and self.is_grid:
             return batch(self._geometry) & self.resolution & non_dual(self._values).without(self.resolution) & self._geometry.shape['vector']
-        return self._geometry.shape.non_channel & self._values
+        return self._geometry.shape.without('vector') & self._values
 
     @property
     def resolution(self):