formatting

examples/0_getting_started.ipynb

@@ -70,15 +70,10 @@
     ")\n",
     "\n",
     "# Initialize the Graph Converter, with dataset, labels and settings\n",
-    "converter = GraphConverter(\n",
-    "    dataset=kloppy_dataset,\n",
-    "    labels=dummy_labels(kloppy_dataset)\n",
-    ")\n",
+    "converter = GraphConverter(dataset=kloppy_dataset, labels=dummy_labels(kloppy_dataset))\n",
     "\n",
     "# Compute the graphs and add them to the CustomSpektralDataset\n",
-    "dataset = CustomSpektralDataset(\n",
-    "    data=converter.to_spektral_graphs()\n",
-    ")"
+    "dataset = CustomSpektralDataset(data=converter.to_spektral_graphs())"
    ]
   },
   {
@@ -91,9 +86,7 @@
     "\n",
     "N, F, S, n_out, n = dataset.dimensions()\n",
     "\n",
-    "train, test = dataset.split_test_train(\n",
-    "    split_train=4, split_test=1, random_seed=42\n",
-    ")\n",
+    "train, test = dataset.split_test_train(split_train=4, split_test=1, random_seed=42)\n",
     "\n",
     "loader_tr = DisjointLoader(train, batch_size=16, epochs=150)\n",
     "loader_te = DisjointLoader(test, batch_size=16, epochs=1, shuffle=False)"

examples/1_tutorial_graph_converter.ipynb

@@ -308,6 +308,7 @@
    "source": [
     "import pickle\n",
     "\n",
+    "\n",
     "def load_pickle(file_path):\n",
     "    with open(file_path, \"rb\") as file:\n",
     "        # Deserialize the object from the file\n",

unravel/soccer/graphs/objects/custom_spektral_dataset.py

@@ -148,7 +148,9 @@ def split_test_train_validation(
             num_test = dataset_length - num_train
             num_validation = 0
 
-        unique_graph_ids = set([g.get("id") if hasattr(g, "id") else None for g in self])
+        unique_graph_ids = set(
+            [g.get("id") if hasattr(g, "id") else None for g in self]
+        )
         if unique_graph_ids == {None}:
             by_graph_id = False