refactor: formatter and linter all files

Author: galogm

graph_datasets/datasets/linkx.py

@@ -1,7 +1,7 @@
 """Datasets from the paper `LINKX <https://github.com/CUAI/Non-Homophily-Large-Scale>`_.
 """
 
-# pylint:disable=invalid-name,duplicate-code
+# pylint:disable=invalid-name,duplicate-code,too-many-branches
 import os
 from typing import Tuple
 
@@ -49,7 +49,7 @@ def even_quantile_labels(vals, n_classes, verbosity: int = 0):
     if verbosity and verbosity > 1:
         logger.info("Generated Class Label Intervals:")
         for class_idx, interval in enumerate(interval_lst):
-            logger.info(f"Class {class_idx}: [{interval[0]}, {interval[1]})]")
+            logger.info("%s", f"Class {class_idx}: [{interval[0]}, {interval[1]})]")
     return label
 
 
@@ -138,6 +138,8 @@ def load_linkx_data(
             directory=directory,
             verbosity=verbosity,
         )
+    else:
+        raise ValueError(f"{dataset_name} in  LINKX not supported.")
 
     if verbosity and verbosity > 1:
         print_dataset_info(
@@ -284,9 +286,8 @@ def load_twitch_gamers_data(
     one_hot = {k: v for v, k in enumerate(nodes["language"].unique())}
     nodes["language"] = [one_hot[lang] for lang in nodes["language"]]
 
-    if task is not None:
-        label = torch.tensor(nodes[task].to_numpy())
-        features = torch.tensor(nodes.drop(task, axis=1).to_numpy(), dtype=torch.float)
+    label = torch.tensor(nodes[task].to_numpy())
+    features = torch.tensor(nodes.drop(task, axis=1).to_numpy(), dtype=torch.float)
 
     if normalize:
         features = features - features.mean(dim=0, keepdim=True)

graph_datasets/datasets/ogb.py

@@ -1,6 +1,7 @@
 """Datasets from `OGB <https://github.com/snap-stanford/ogb>`_.
 """
 
+# pylint: disable=invalid-name,
 import traceback
 from typing import Tuple
 
@@ -59,7 +60,7 @@ def load_ogb_data(
 
     try:
         splits = dataset.get_idx_split()
-    except Exception as _:
+    except RuntimeError as _:
         traceback.print_exc()
         splits = None
 
@@ -98,7 +99,7 @@ def load_ogb_data(
 def to_sparse_tensor(edge_index, edge_feat, num_nodes):
     """converts the edge_index into SparseTensor"""
     num_edges = edge_index.size(1)
-    (row, col), N, E = edge_index, num_nodes, num_edges
+    (row, col), N, _ = edge_index, num_nodes, num_edges
     perm = (col * N + row).argsort()
     row, col = row[perm], col[perm]
     value = edge_feat[perm]

graph_datasets/load_data.py

@@ -1,7 +1,7 @@
 """Load Graph Datasets
 """
 
-# pylint:disable=protected-access
+# pylint:disable=protected-access,too-many-locals,too-many-branches,too-many-statements,
 import ssl
 from typing import Tuple, Union
 
@@ -188,8 +188,12 @@ def load_data(
     graph.directed = not to_simple
 
     if verbosity:
+        dm = (
+            f"{source.upper()} undirected {dataset_name}\n"
+            f"add_self_loop={add_self_loop} rm_self_loop={rm_self_loop}"
+        )
         print_dataset_info(
-            dataset_name=f"{source.upper()} undirected {dataset_name}\n  add_self_loop={add_self_loop} rm_self_loop={rm_self_loop}",
+            dataset_name=dm,
             n_nodes=graph.num_nodes(),
             n_edges=graph.num_edges(),
             n_feats=graph.ndata["feat"].shape[1],

graph_datasets/utils/common.py

@@ -71,7 +71,7 @@ def tab_printer(
         table.set_cols_dtype(cols_dtype)
     table.add_rows(params)
 
-    logger.info(f"\n{table.draw()}")
+    logger.info("%s", f"\n{table.draw()}")
 
     return table.draw()
 

graph_datasets/utils/logging.py

@@ -1,6 +1,7 @@
 """Logger: Adapted from the optuna.logging.
 """
 
+# pylint: disable=global-statement,
 from __future__ import annotations
 
 import logging

tests/test.py

@@ -1,6 +1,8 @@
 """Test
 """
 
+import sys
+
 # pylint:disable=duplicate-code
 from graph_datasets import load_data
 from graph_datasets.data_info import DATASETS
@@ -65,4 +67,4 @@ def main(_source, _dataset):
     for source, datasets in DATASETS.items():
         for dataset in datasets:
             main(source, dataset)
-            exit(1)
+            sys.exit(1)