fix: add host to client

picselliahq · Dec 11, 2024 · 7f64f94 · 7f64f94
1 parent 07369b5
commit 7f64f94
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/pipelines/datalake_autotagging/create_datalake_autotagging_processing.py b/src/pipelines/datalake_autotagging/create_datalake_autotagging_processing.py
@@ -37,6 +37,7 @@ def create_processing(
     parser = ArgumentParser("Create a processing")
     parser.add_argument("--api_token", type=str)
     parser.add_argument("--organization_id", type=str)
+    parser.add_argument("--host", type=str)
     parser.add_argument("--processing_name", type=str)
     parser.add_argument("--processing_type", type=str)
     parser.add_argument("--default_cpu", type=int)
@@ -46,7 +47,9 @@ def create_processing(
     parser.add_argument("--docker_flags", nargs="+", type=str, default=None)
     args = parser.parse_args()
 
-    client = Client(api_token=args.api_token, organization_id=args.organization_id)
+    client = Client(
+        api_token=args.api_token, organization_id=args.organization_id, host=args.host
+    )
 
     default_parameters = {
         "tags_list": ["is_woman", "is_man"],

diff --git a/src/pipelines/datalake_autotagging/launch_datalake_autotagging_processing.py b/src/pipelines/datalake_autotagging/launch_datalake_autotagging_processing.py
@@ -50,6 +50,7 @@ def launch_processing(
     parser = ArgumentParser("Launch a processing")
     parser.add_argument("--api_token", type=str)
     parser.add_argument("--organization_id", type=str)
+    parser.add_argument("--host", type=str)
     parser.add_argument("--processing_name", type=str)
     parser.add_argument("--datalake_id", type=str)
     parser.add_argument("--target_datalake_name", type=str, default=None)
@@ -62,7 +63,9 @@ def launch_processing(
     parser.add_argument("--batch_size", type=int, default=8)
     args = parser.parse_args()
 
-    client = Client(api_token=args.api_token, organization_id=args.organization_id)
+    client = Client(
+        api_token=args.api_token, organization_id=args.organization_id, host=args.host
+    )
 
     data_ids = [UUID(data_id) for data_id in args.data_ids]