fix pre_transform in InMemoryDataset

pyg-team · Dec 2, 2020 · bcdb323 · bcdb323
1 parent 26335c3
commit bcdb323
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/torch_geometric/data/in_memory_dataset.py b/torch_geometric/data/in_memory_dataset.py
@@ -67,12 +67,13 @@ def len(self):
         return 0
 
     def get(self, idx):
-        if self.__data_list__ is None:
-            self.__data_list__ = self.len() * [None]
-        else:
-            data = self.__data_list__[idx]
-            if data is not None:
-                return copy.copy(data)
+        if hasattr(self, '__data_list__'):
+            if self.__data_list__ is None:
+                self.__data_list__ = self.len() * [None]
+            else:
+                data = self.__data_list__[idx]
+                if data is not None:
+                    return copy.copy(data)
 
         data = self.data.__class__()
         if hasattr(self.data, '__num_nodes__'):
@@ -90,7 +91,9 @@ def get(self, idx):
                 s = slice(start, end)
             data[key] = item[s]
 
-        self.__data_list__[idx] = copy.copy(data)
+        if hasattr(self, '__data_list__'):
+            self.__data_list__[idx] = copy.copy(data)
+
         return data
 
     @staticmethod