linting

pyt-team · Jul 9, 2024 · fa9f996 · fa9f996
1 parent e1b4ac3
commit fa9f996
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/modules/transforms/liftings/graph2simplicial/directed_clique_lifting.py b/modules/transforms/liftings/graph2simplicial/directed_clique_lifting.py
@@ -2,8 +2,8 @@
 
 import networkx as nx
 import torch_geometric
-from torch_geometric.utils.undirected import is_undirected
 from toponetx.classes import SimplicialComplex
+from torch_geometric.utils.undirected import is_undirected
 
 from modules.transforms.liftings.graph2simplicial.base import Graph2SimplicialLifting
 

diff --git a/test/transforms/liftings/graph2simplicial/test_directed_clique_lifting.py b/test/transforms/liftings/graph2simplicial/test_directed_clique_lifting.py
@@ -1,8 +1,8 @@
 """Test the message passing module."""
 
+import networkx as nx
 import torch
 import torch_geometric
-import networkx as nx
 
 from modules.transforms.liftings.graph2simplicial.directed_clique_lifting import (
     DirectedSimplicialCliqueLifting,

diff --git a/tutorials/graph2simplicial/directed_clique_lifting.ipynb b/tutorials/graph2simplicial/directed_clique_lifting.ipynb
@@ -55,6 +55,10 @@
     "# With this cell any imported module is reloaded before each cell execution\n",
     "%load_ext autoreload\n",
     "%autoreload 2\n",
+    "import networkx as nx\n",
+    "import torch\n",
+    "import torch_geometric\n",
+    "\n",
     "from modules.data.load.loaders import GraphLoader\n",
     "from modules.data.preprocess.preprocessor import PreProcessor\n",
     "from modules.transforms.liftings.graph2simplicial.directed_clique_lifting import (\n",
@@ -65,10 +69,7 @@
     "    load_dataset_config,\n",
     "    load_model_config,\n",
     "    load_transform_config,\n",
-    ")\n",
-    "import networkx as nx\n",
-    "import torch\n",
-    "import torch_geometric"
+    ")"
    ]
   },
   {