Use black --config ../.../black.toml .../snowstorm_dataset.py

Author: christianlocatelli

src/graphnet/datasets/snowstorm_dataset.py

@@ -159,7 +159,9 @@ def _prepare_args(
         train_val_selection = []
         test_selection = []
         for path in dataset_paths:
-            train_val_selection.append(train_val["event_no"].get_group(path).tolist())
+            train_val_selection.append(
+                train_val["event_no"].get_group(path).tolist()
+            )
             test_selection.append(test["event_no"].get_group(path).tolist())
 
         dataset_args = {
@@ -188,7 +190,9 @@ def _create_comment(cls, event_counts: Dict[str, int] = {}) -> None:
             runid_string += f"RunID {k} contains {v:10d} events\n"
             tot += v
         cls._comments = (
-            f"Contains ~{tot/1e6:.1f} million events:\n" + runid_string + fixed_string
+            f"Contains ~{tot / 1e6:.1f} million events:\n"
+            + runid_string
+            + fixed_string
         )
 
     def _get_dir_name(self, source_file_path: str) -> str: