fix(data): make FeatureStore and GraphStore proper abstract base …

…classes (#9002)

Co-authored-by: rusty1s <matthias.fey@tu-dortmund.de>

torch_geometric/data/feature_store.py

@@ -21,7 +21,7 @@
 * Async `put` and `get` functionality
 """
 import copy
-from abc import abstractmethod
+from abc import ABC, abstractmethod
 from dataclasses import dataclass
 from enum import Enum
 from typing import Any, List, Optional, Tuple, Union
@@ -262,7 +262,7 @@ def __repr__(self) -> str:
 # libraries use customized logic during mini-batch for `Mapping` base classes.
 
 
-class FeatureStore:
+class FeatureStore(ABC):
     r"""An abstract base class to access features from a remote feature store.
 
     Args:

torch_geometric/data/graph_store.py

@@ -17,7 +17,7 @@
 the graph in interesting manners based on the provided metadata.
 """
 import copy
-from abc import abstractmethod
+from abc import ABC, abstractmethod
 from collections import defaultdict
 from dataclasses import dataclass
 from enum import Enum
@@ -98,7 +98,7 @@ def __init__(
         self.size = size
 
 
-class GraphStore:
+class GraphStore(ABC):
     r"""An abstract base class to access edges from a remote graph store.
 
     Args:

torch_geometric/datasets/entities.py

@@ -89,10 +89,10 @@ def __init__(
         assert self.name in ['aifb', 'am', 'mutag', 'bgs']
         super().__init__(root, transform, pre_transform,
                          force_reload=force_reload)
-        self.load(
-            self.processed_paths[0],
-            data_cls=HeteroData if hetero else Data,
-        )
+        if hetero:
+            self.load(self.processed_paths[0], data_cls=HeteroData)
+        else:
+            self.load(self.processed_paths[0], data_cls=Data)
 
     @property
     def raw_dir(self) -> str: