Merge pull request #133 from josephenguehard/refactor_getitem

Refactor __get_item__ into __getitem__
benedekrozemberczki · Feb 2, 2022 · 5d5c7dd · 5d5c7dd
2 parents 90fc89e + 83830d9
commit 5d5c7dd
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 12 deletions.
diff --git a/torch_geometric_temporal/signal/dynamic_graph_static_signal.py b/torch_geometric_temporal/signal/dynamic_graph_static_signal.py
@@ -105,7 +105,7 @@ def _get_additional_features(self, time_index: int):
     def __len__(self):
         return len(self.targets)
 
-    def __get_item__(self, time_index: int):
+    def __getitem__(self, time_index: int):
         x = self._get_feature()
         edge_index = self._get_edge_index(time_index)
         edge_weight = self._get_edge_weight(time_index)
@@ -118,7 +118,7 @@ def __get_item__(self, time_index: int):
 
     def __next__(self):
         if self.t < len(self.targets):
-            snapshot = self.__get_item__(self.t)
+            snapshot = self[self.t]
             self.t = self.t + 1
             return snapshot
         else:

diff --git a/torch_geometric_temporal/signal/dynamic_graph_static_signal_batch.py b/torch_geometric_temporal/signal/dynamic_graph_static_signal_batch.py
@@ -116,7 +116,7 @@ def _get_additional_features(self, time_index: int):
         }
         return additional_features
 
-    def __get_item__(self, time_index: int):
+    def __getitem__(self, time_index: int):
         x = self._get_feature()
         edge_index = self._get_edge_index(time_index)
         edge_weight = self._get_edge_weight(time_index)
@@ -130,7 +130,7 @@ def __get_item__(self, time_index: int):
 
     def __next__(self):
         if self.t < len(self.targets):
-            snapshot = self.__get_item__(self.t)
+            snapshot = self[self.t]
             self.t = self.t + 1
             return snapshot
         else:

diff --git a/torch_geometric_temporal/signal/dynamic_graph_temporal_signal.py b/torch_geometric_temporal/signal/dynamic_graph_temporal_signal.py
@@ -105,7 +105,7 @@ def _get_additional_features(self, time_index: int):
         }
         return additional_features
 
-    def __get_item__(self, time_index):
+    def __getitem__(self, time_index):
         x = self._get_features(time_index)
         edge_index = self._get_edge_index(time_index)
         edge_weight = self._get_edge_weight(time_index)
@@ -118,7 +118,7 @@ def __get_item__(self, time_index):
 
     def __next__(self):
         if self.t < len(self.features):
-            snapshot = self.__get_item__(self.t)
+            snapshot = self[self.t]
             self.t = self.t + 1
             return snapshot
         else:

diff --git a/torch_geometric_temporal/signal/dynamic_graph_temporal_signal_batch.py b/torch_geometric_temporal/signal/dynamic_graph_temporal_signal_batch.py
@@ -119,7 +119,7 @@ def _get_additional_features(self, time_index: int):
         }
         return additional_features
 
-    def __get_item__(self, time_index: int):
+    def __getitem__(self, time_index: int):
         x = self._get_feature(time_index)
         edge_index = self._get_edge_index(time_index)
         edge_weight = self._get_edge_weight(time_index)
@@ -133,7 +133,7 @@ def __get_item__(self, time_index: int):
 
     def __next__(self):
         if self.t < len(self.features):
-            snapshot = self.__get_item__(self.t)
+            snapshot = self[self.t]
             self.t = self.t + 1
             return snapshot
         else:

diff --git a/torch_geometric_temporal/signal/static_graph_temporal_signal.py b/torch_geometric_temporal/signal/static_graph_temporal_signal.py
@@ -100,7 +100,7 @@ def _get_additional_features(self, time_index: int):
         }
         return additional_features
 
-    def __get_item__(self, time_index: int):
+    def __getitem__(self, time_index: int):
         x = self._get_features(time_index)
         edge_index = self._get_edge_index()
         edge_weight = self._get_edge_weight()
@@ -113,7 +113,7 @@ def __get_item__(self, time_index: int):
 
     def __next__(self):
         if self.t < len(self.features):
-            snapshot = self.__get_item__(self.t)
+            snapshot = self[self.t]
             self.t = self.t + 1
             return snapshot
         else:

diff --git a/torch_geometric_temporal/signal/static_graph_temporal_signal_batch.py b/torch_geometric_temporal/signal/static_graph_temporal_signal_batch.py
@@ -110,7 +110,7 @@ def _get_additional_features(self, time_index: int):
         }
         return additional_features
 
-    def __get_item__(self, time_index: int):
+    def __getitem__(self, time_index: int):
         x = self._get_feature(time_index)
         edge_index = self._get_edge_index()
         edge_weight = self._get_edge_weight()
@@ -124,7 +124,7 @@ def __get_item__(self, time_index: int):
 
     def __next__(self):
         if self.t < len(self.features):
-            snapshot = self.__get_item__(self.t)
+            snapshot = self[self.t]
             self.t = self.t + 1
             return snapshot
         else: