chore: use black formatter

options/predict_options.py

@@ -15,28 +15,18 @@ def initialize(self, parser):
             "--model-in-file",
             type=str,
             help="file path to generator model (.pth file)",
-            required=True
+            required=True,
         )
 
         parser.add_argument(
-            "--img-in",
-            type=str,
-            help="image to transform",
-            required=True
+            "--img-in", type=str, help="image to transform", required=True
         )
 
         parser.add_argument(
-            "--img-out",
-            type=str,
-            help="transformed image",
-            required=True
+            "--img-out", type=str, help="transformed image", required=True
         )
 
-        parser.add_argument(
-            "--cpu",
-            action="store_true",
-            help="whether to use CPU"
-        )
+        parser.add_argument("--cpu", action="store_true", help="whether to use CPU")
 
         parser.add_argument(
             "--gpuid",

scripts/gen_single_image.py

@@ -44,6 +44,7 @@ def load_model(modelpath, model_in_file, cpu, gpuid):
     model = model.to(device)
     return model, opt, device
 
+
 def launch_predict(args):
     # loading model
     modelpath = args.model_in_file.replace(os.path.basename(args.model_in_file), "")
@@ -97,7 +98,9 @@ def launch_predict(args):
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument(
-        "--model-in-file", help="file path to generator model (.pth file)", required=True
+        "--model-in-file",
+        help="file path to generator model (.pth file)",
+        required=True,
     )
 
     parser.add_argument("--img-in", help="image to transform", required=True)

server/joligen_api.py

@@ -18,6 +18,7 @@
 from options.predict_options import PredictOptions
 
 import sys
+
 sys.path.append(os.path.join(os.path.dirname(__file__), "../scripts"))
 from gen_single_image import launch_predict
 
@@ -128,6 +129,7 @@ def stop_training(process):
 def is_alive(process):
     return process.is_alive()
 
+
 @app.post(
     "/predict",
     status_code=201,
@@ -154,6 +156,7 @@ async def predict(request: Request):
 
     return {"message": "ok", "name": name, "status": "running"}
 
+
 @app.post(
     "/train/{name}",
     status_code=201,