fix: minor bug when only space id is provided for WML inference (#1583)

fix: wml inference with space id only Signed-off-by: Cheng Qian <[email protected]> Co-authored-by: Cheng Qian <[email protected]>
IBM · Feb 9, 2025 · ed773e9 · ed773e9
1 parent bf2d511
commit ed773e9
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: