fix: wml inference with space id only

Signed-off-by: Cheng Qian <[email protected]>
IBM · Feb 7, 2025 · 654530b · 654530b
1 parent bf2d511
commit 654530b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/unitxt/inference.py b/src/unitxt/inference.py
@@ -2004,6 +2004,8 @@ def _read_wml_credentials_from_env() -> CredentialsWML:
                 "only one of those defined in the env."
             )
             credentials["space_id"] = space_id
+        elif space_id:
+            credentials["space_id"] = space_id
         elif project_id:
             credentials["project_id"] = project_id
         else: