black format

wells-wood-research · Dec 20, 2023 · 56c35a2 · 56c35a2
1 parent 07e9049
commit 56c35a2
Showing 1 changed file with 18 additions and 5 deletions.
diff --git a/src/aposteriori/data_prep/cli.py b/src/aposteriori/data_prep/cli.py
@@ -113,7 +113,18 @@
 @click.option(
     "-ae",
     "--atom_encoder",
-    type=click.Choice(["CNO", "CNOCB", "CNOCACB", "CNOCBCA", "CNOCACBQ", "CNOCBCAQ", "CNOCACBP", "CNOCBCAP"]),
+    type=click.Choice(
+        [
+            "CNO",
+            "CNOCB",
+            "CNOCACB",
+            "CNOCBCA",
+            "CNOCACBQ",
+            "CNOCBCAQ",
+            "CNOCACBP",
+            "CNOCBCAP",
+        ]
+    ),
     default="CNO",
     required=True,
     help=(
@@ -269,14 +280,14 @@ def cli(
         "CNOCB": Codec.CNOCB,
         "CNOCACB": Codec.CNOCACB,
         "CNOCACBQ": Codec.CNOCACBQ,
-        "CNOCACBP": Codec.CNOCACBP
+        "CNOCACBP": Codec.CNOCACBP,
     }
 
     # List of deprecated encodings and their replacements
     deprecated_encodings = {
         "CNOCBCA": "CNOCACB",
         "CNOCBCAQ": "CNOCACBQ",
-        "CNOCBCAP": "CNOCACBP"
+        "CNOCBCAP": "CNOCACBP",
     }
 
     # Create Codec based on atom_encoder
@@ -285,8 +296,10 @@ def cli(
     elif atom_encoder in deprecated_encodings:
         replacement = deprecated_encodings[atom_encoder]
         codec = current_codec_mapping[replacement]()
-        warnings.warn(f"{atom_encoder} encoding is deprecated and will be removed in future versions, "
-                    f"atoms will be encoded as {replacement}")
+        warnings.warn(
+            f"{atom_encoder} encoding is deprecated and will be removed in future versions, "
+            f"atoms will be encoded as {replacement}"
+        )
 
     make_frame_dataset(
         structure_files=structure_files,