diff --git a/phi/field/_field.py b/phi/field/_field.py index 1fbff483c..67dffdc0d 100644 --- 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):