🔁 Merge pull request #120 from bfGraph/develop

🔁 Update official-docs with develop branch

Author: nithinmanoj10

.github/workflows/pytest.yaml

@@ -0,0 +1,28 @@
+name: Run Unit Test via Pytest  
+  
+on: [push]  
+  
+jobs:  
+  build:  
+    runs-on: ubuntu-latest  
+    strategy:  
+      matrix:  
+        python-version: ["3.8"]  
+  
+    steps:  
+      - uses: actions/checkout@v3  
+      - name: Set up Python ${{ matrix.python-version }}  
+        uses: actions/setup-python@v4  
+        with:  
+          python-version: ${{ matrix.python-version }}  
+      - name: Install dependencies  
+        run: |  
+          python -m pip install --upgrade pip  
+          pip install -e .[dev]
+          pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+      - name: Test with pytest  
+        run: |  
+          coverage run -m pytest  -v -s -p no:warnings
+      - name: Generate Coverage Report  
+        run: |  
+          coverage report -m

.gitignore

@@ -30,4 +30,7 @@ dist
 *.npy
 .coverage
 dev-stgraph/
-htmlconv/
+htmlconv/
+*.txt
+egl_kernel.cu
+egl_kernel.ptx

benchmarking/gcn/seastar/utils.py

@@ -1,21 +1,32 @@
 import torch
 
+
 def accuracy(logits, labels):
     _, indices = torch.max(logits, dim=1)
     correct = torch.sum(indices == labels)
     return correct.item() * 1.0 / len(labels)
 
+
 # GPU | CPU
 def get_default_device():
-    
     if torch.cuda.is_available():
-        return torch.device('cuda:0')
+        return torch.device("cuda:0")
     else:
-        return torch.device('cpu')
+        return torch.device("cpu")
+
 
 def to_default_device(data):
-    
-    if isinstance(data,(list,tuple)):
-        return [to_default_device(x,get_default_device()) for x in data]
-    
-    return data.to(get_default_device(),non_blocking = True)
+    if isinstance(data, (list, tuple)):
+        return [to_default_device(x, get_default_device()) for x in data]
+
+    return data.to(get_default_device(), non_blocking=True)
+
+
+def generate_train_mask(size: int, train_test_split: int) -> list:
+    cutoff = size * train_test_split
+    return [1 if i < cutoff else 0 for i in range(size)]
+
+
+def generate_test_mask(size: int, train_test_split: int) -> list:
+    cutoff = size * train_test_split
+    return [0 if i < cutoff else 1 for i in range(size)]

pyproject.toml

@@ -44,7 +44,8 @@ dependencies = [
 [project.optional-dependencies]
 dev = [
     "black",
-    "pytest",
+    "pytest >= 7.4.3",
+    "pytest-cov >= 4.1.0",
     "tqdm >= 4.64.1",
     "build >= 0.10.0",
     "gdown >= 4.6.6",

requirements.txt

@@ -1,5 +1,5 @@
 cuda_python==12.1.0
-Jinja2==3.1.2
+Jinja2==3.1.3
 networkx==3.1
 numpy==1.23.4
 pandas==1.3.5
@@ -10,7 +10,7 @@ snoop==0.4.2
 sphinx_rtd_theme==1.2.0
 sympy==1.11.1
 termcolor==2.3.0
-tqdm==4.64.1
+tqdm==4.66.3
 pybind11==2.10.4
 build==0.10.0 
 pynvml==11.5.0

stgraph/benchmark_tools/table.py

@@ -1,25 +1,30 @@
+from __future__ import annotations
+
 from rich.console import Console
 from rich.table import Table
 
-console = Console()
 
 class BenchmarkTable:
     def __init__(self, title: str, col_name_list: list[str]):
-        self.title = '\n' + title + '\n'
+        self.title = "\n" + title + "\n"
         self.col_name_list = col_name_list
         self._table = Table(title=self.title, show_edge=False, style="black bold")
         self._num_cols = len(col_name_list)
         self._num_rows = 0
-        
+
         self._table_add_columns()
-        
+
     def _table_add_columns(self):
         for col_name in self.col_name_list:
             self._table.add_column(col_name, justify="left")
-            
+
     def add_row(self, values: list):
         values_str = tuple([str(val) for val in values])
         self._table.add_row(*values_str)
-    
-    def display(self):
-        console.print(self._table)
+
+    def display(self, output_file=None):
+        if not output_file:
+            console = Console()
+        else:
+            console = Console(file=output_file)
+        console.print(self._table)

stgraph/dataset/dynamic/england_covid_dataloader.py

@@ -53,6 +53,7 @@ class EnglandCovidDataLoader(STGraphDynamicDataset):
         The name of the dataset.
     gdata : dict
         Graph meta data.
+
     """
 
     def __init__(
@@ -65,6 +66,16 @@ def __init__(
         """COVID-19 cases in England's NUTS3 regions."""
         super().__init__()
 
+        if not isinstance(lags, int):
+            raise TypeError("lags must be of type int")
+        if lags < 0:
+            raise ValueError("lags must be a positive integer")
+
+        if cutoff_time is not None and not isinstance(cutoff_time, int):
+            raise TypeError("cutoff_time must be of type int")
+        if cutoff_time is not None and cutoff_time < 0:
+            raise ValueError("cutoff_time must be a positive integer")
+
         self.name = "England_COVID"
         self._url = "https://raw.githubusercontent.com/benedekrozemberczki/pytorch_geometric_temporal/master/dataset/england_covid.json"
         self._verbose = verbose

stgraph/dataset/static/cora_dataloader.py

@@ -61,6 +61,7 @@ class CoraDataLoader(STGraphStaticDataset):
         The name of the dataset.
     gdata : dict
         Graph meta data.
+
     """
 
     def __init__(

stgraph/dataset/stgraph_dataset.py

@@ -67,6 +67,7 @@ def __init__(self: STGraphDataset) -> None:
 
         _load_dataset()
         Loads the dataset from cache
+
         """
         self.name = ""
         self.gdata = {}
@@ -106,6 +107,7 @@ def _has_dataset_cache(self: STGraphDataset) -> bool:
                 # The dataset is cached, continue cached operations
             else:
                 # The dataset is not cached, continue load and save operations
+
         """
         user_home_dir = os.path.expanduser("~")
         stgraph_dir = user_home_dir + "/.stgraph"
@@ -128,6 +130,7 @@ def _get_cache_file_path(self: STGraphDataset) -> str:
         -------
         str
             The absolute path of the cached dataset file
+
         """
         user_home_dir = os.path.expanduser("~")
         stgraph_dir = user_home_dir + "/.stgraph"

stgraph/dataset/temporal/hungarycp_dataloader.py

@@ -58,6 +58,7 @@ class HungaryCPDataLoader(STGraphTemporalDataset):
         The name of the dataset.
     gdata : dict
         Graph meta data.
+
     """
 
     def __init__(