fix: change argument name to remove raised exception

jolibrain · Aug 22, 2023 · 439099f · 439099f
1 parent 38bc932
commit 439099f
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/options/predict_options.py b/options/predict_options.py
@@ -231,16 +231,8 @@ def initialize(self, parser):
             help="eta parameter for ddim variance",
         )
 
-        # TODO: raise exception when running server
-        #   File "/home/alx/code/alx/joliGEN/server/joligen_api.py", line 91, in <module>
-        #  PredictBodySchema["properties"]["predict_options"] = PredictOptions().get_schema()
-        #   File "/home/alx/code/alx/joliGEN/options/base_options.py", line 1261, in get_schema
-        #     json_vals = self.to_json()
-        #   File "/home/alx/code/alx/joliGEN/options/base_options.py", line 965, in to_json
-        #     parent[path[-1]] = v
-        # TypeError: 'bool' object does not support item assignment
-        parser.add_argument(
-            "--cls",
+        parser.add_argument(
+            "--cls-override",
             action="store_false",
             help="override input bbox classe for generation",
         )