diff --git a/benchmark/trainer/load_data.py b/benchmark/trainer/load_data.py
index 70415d5..528e394 100755
--- a/benchmark/trainer/load_data.py
+++ b/benchmark/trainer/load_data.py
@@ -174,7 +174,7 @@ def _resolve_import(self, args: Namespace) -> Tuple[str, str, dict]:
else:
self.metric = 's_f1i'
if self.data not in ['snap-patents']:
- self._T_insert(T.ToUndirected(), index=0)
+ kwargs['transform'] = self._T_insert(T.ToUndirected(), index=0)
elif self.data in ['penn94', 'amherst41', 'cornell5', 'johns_hopkins55', 'reed98']:
module_name = 'dataset_process'
class_name = 'FB100'
@@ -202,7 +202,7 @@ def _resolve_import(self, args: Namespace) -> Tuple[str, str, dict]:
kwargs = dict(
root=DATAPATH.joinpath('PyG'),
name=self.data,
- transform=self.transform,)
+ transform=self._T_insert(T.ToUndirected(), index=0),)
pyg_mapping = {
'cora': 'Planetoid',
'citeseer': 'Planetoid',
diff --git a/benchmark/utils/logger.py b/benchmark/utils/logger.py
index 4756ecc..7e807c5 100755
--- a/benchmark/utils/logger.py
+++ b/benchmark/utils/logger.py
@@ -219,6 +219,7 @@ def merge(self,
logger.data.index = rows
if suffix:
logger.data.columns = [f'{coli}_{suffix}' for coli in logger.data.columns]
+ logger.fmt.index = [f'{coli}_{suffix}' for coli in logger.fmt.index]
self._set(logger.data, logger.fmt)
return self
@@ -336,3 +337,6 @@ def get_str(self,
def __str__(self) -> str:
r"""String for print on screen."""
return self.get_str(maxlen=80)
+
+ def __repr__(self) -> str:
+ return self.get_str()
diff --git a/docs/source/index.rst b/docs/source/index.rst
index b173ea6..2f1f7fe 100644
--- a/docs/source/index.rst
+++ b/docs/source/index.rst
@@ -9,7 +9,7 @@ pyg_spectral
| 🔍 `Documentation `__
| 👾 `GitHub `__ |gh-bn|
- | 📄 `Paper `__
+ | 📄 `Paper `__
| 📎 `Citation `__
``pyg_spectral`` is a `PyG `_-based framework for analyzing, implementing, and benchmarking spectral GNNs with effectiveness and efficiency evaluations.