[geom,field] Update variable/value attrs

tum-pbs · Mar 11, 2024 · ab75193 · ab75193
1 parent a2d774f
commit ab75193
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git 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
@@ -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)