From ab7519342a495b1241ea152c15eedde710a25ab6 Mon Sep 17 00:00:00 2001 From: Philipp Holl Date: Mon, 11 Mar 2024 20:14:24 +0100 Subject: [PATCH] [geom,field] Update variable/value attrs --- phi/field/_field.py | 4 ++-- phi/geom/_mesh.py | 4 +++- 2 files changed, 5 insertions(+), 3 deletions(-) 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)