feat: separate graph_utils from graph_datasets

Author: galogm

doc_repo

@@ -1,11 +1,3 @@
-evaluation
-======================
-.. automodule:: graph_datasets.utils.evaluation.evaluation
-   :members:
-   :show-inheritance:
-   :exclude-members:
-   :no-undoc-members:
-
 statistics
 ======================
 .. automodule:: graph_datasets.utils.statistics
@@ -14,27 +6,6 @@ statistics
    :exclude-members: get_long_edges, get_same_class_deg
    :no-undoc-members:
 
-model management
-======================
-.. automodule:: graph_datasets.utils.model_management
-   :members:
-   :show-inheritance:
-   :no-undoc-members:
-
-output
-======================
-.. automodule:: graph_datasets.utils.output
-   :members:
-   :show-inheritance:
-   :no-undoc-members:
-
-plt
-======================
-.. automodule:: graph_datasets.utils.plt
-   :members:
-   :show-inheritance:
-   :no-undoc-members:
-
 common
 ======================
 .. automodule:: graph_datasets.utils.common

docs/rst/graph_datasets.utils.rst

@@ -134,7 +134,7 @@ def load_linkx_data(
 
     if verbosity and verbosity > 1:
         print_dataset_info(
-            dataset_name=f"LINKX Original {dataset_name}",
+            dataset_name=f"LINKX original {dataset_name}",
             n_nodes=g.num_nodes(),
             n_edges=g.num_edges(),
             n_feats=g.ndata["feat"].shape[1],

graph_datasets/datasets/linkx.py

@@ -137,7 +137,7 @@ def load_data(
 
     if verbosity:
         print_dataset_info(
-            dataset_name=f"{source} undirected {dataset_name}\nwithout self-loops",
+            dataset_name=f"{source.upper()} undirected {dataset_name}\nwithout self-loops",
             n_nodes=graph.num_nodes(),
             n_edges=graph.num_edges(),
             n_feats=graph.ndata["feat"].shape[1],

graph_datasets/load_data.py

@@ -2,18 +2,6 @@
 """
 # pylint:disable=invalid-name
 from .common import *
-from .evaluation import evaluate_from_embed_file
-from .model_management import check_modelfile_exists
-from .model_management import get_modelfile_path
-from .model_management import load_model
-from .model_management import save_model
-from .model_management import set_device
-from .model_management import set_seed
-from .output import csv2file
-from .output import make_parent_dirs
-from .output import refresh_file
-from .output import save_to_csv_files
-from .plt import draw_chart
 from .statistics import edge_homo
 from .statistics import node_homo
 from .statistics import statistics

graph_datasets/utils/__init__.py

@@ -2,53 +2,14 @@
 """
 import os
 import sys
-from datetime import datetime
 from typing import Any
 from typing import Dict
 from typing import List
 
 import gdown
-import pytz
 from texttable import Texttable
 
 
-def format_result(
-    dataset: str,
-    source: str,
-    model: str,
-    sort_kw: bool = True,
-    timezone="Asia/Shanghai",
-    **kwargs,
-):
-    if sort_kw:
-        kwargs = dict(sorted(kwargs.items()))
-
-    kwargs.update(
-        {
-            "ds": dataset,
-            "src": source,
-            "model": model,
-            "time": get_str_time(timezone),
-        }
-    )
-    return kwargs
-
-
-def get_str_time(timezone="Asia/Shanghai"):
-    """Return localtime in the format of %Y-%m-%d-%H:%M:%S."""
-
-    # Set the timezone to timezone
-    pytz_timezone = pytz.timezone(timezone)
-
-    # Get the current time in UTC
-    utc_now = datetime.utcnow()
-
-    # Convert UTC time to timezone time
-    pytz_now = utc_now.replace(tzinfo=pytz.utc).astimezone(pytz_timezone)
-
-    return pytz_now.strftime("%Y-%m-%d-%H:%M:%S")
-
-
 def format_value(value) -> Any:
     """Return number as string with comma split.
 
@@ -95,8 +56,7 @@ def tab_printer(
             [
                 k.replace("_", " "),
                 f"{args[k]}" if isinstance(args[k], bool) else format_value(args[k]),
-            ]
-            for k in keys
+            ] for k in keys
         ]
     )
     if cols_align is not None:
@@ -119,9 +79,7 @@ def download_tip(info: Dict) -> None:
         data_file (str): filepath.
         url (str): url for downloading.
     """
-    info[
-        "Tip"
-    ] = "If the download fails, \
+    info["Tip"] = "If the download fails, \
 use the 'Download URL' to download manually and move the file to the 'Save Path'."
 
     tab_printer(info)