diff --git a/phi/field/_field.py b/phi/field/_field.py index 1adaf8f45..9a8b4ab80 100644 --- a/phi/field/_field.py +++ b/phi/field/_field.py @@ -540,10 +540,10 @@ def dimension(self, name: str): return BoundDim(self, name) def __value_attrs__(self): - return '_values', '_boundary' + return '_values', def __variable_attrs__(self): - return '_values', '_geometry' + return '_values', '_geometry', '_boundary' def __expand__(self, dims: Shape, **kwargs) -> 'Field': return self.with_values(expand(self.values, dims, **kwargs)) diff --git a/phi/geom/_mesh.py b/phi/geom/_mesh.py index ec61b7d01..0648d6c71 100644 --- a/phi/geom/_mesh.py +++ b/phi/geom/_mesh.py @@ -73,9 +73,11 @@ def __init__(self, vertices: Graph, # e_face = def __variable_attrs__(self): - return () return '_vertices', '_center', '_volume', '_faces', '_valid_mask', '_face_vertices', '_relative_face_distance', '_neighbor_offsets' + def __value_attrs__(self): + return '_vertices', + @property def shape(self) -> Shape: return shape(self._polygons).non_spatial & channel(self._vertices)