Use class setter properties if present

test/data/test_data.py

@@ -213,3 +213,34 @@ def test_data_share_memory():
     for data in data_list:
         assert data.x.is_shared()
         assert torch.allclose(data.x, torch.full((8, ), 4.))
+
+
+def test_data_subclass_setter():
+    class Graph(Data):
+        def __init__(self):
+            super().__init__()
+            self.my_attr = 1.0
+            self.my_attr2 = 2
+
+        @property
+        def my_attr(self):
+            return self._my_attr
+
+        @my_attr.setter
+        def my_attr(self, value):
+            self._my_attr = int(value)
+
+    g = Graph()
+    assert g._store == {'my_attr2': 2}
+    assert '_my_attr' in g._store.__dict__
+    assert isinstance(g.my_attr, int)
+
+    x = torch.tensor([[1, 2], [3, 4]])
+    g.x = x
+    assert g._store == {'x': x, 'my_attr2': 2}
+
+    g.my_attr = 2.0
+    assert isinstance(g.my_attr, int)
+
+    g.my_attr2 = 'asdf'
+    assert g._store == {'x': x, 'my_attr2': 'asdf'}

torch_geometric/data/data.py

@@ -375,7 +375,11 @@ def __getattr__(self, key: str) -> Any:
         return getattr(self._store, key)
 
     def __setattr__(self, key: str, value: Any):
-        setattr(self._store, key, value)
+        propobj = getattr(self.__class__, key, None)
+        if propobj is None or propobj.fset is None:
+            setattr(self._store, key, value)
+        else:
+            propobj.fset(self, value)
 
     def __delattr__(self, key: str):
         delattr(self._store, key)