From ed773e99daad9965bd2638b00301a2c3ffb78f0a Mon Sep 17 00:00:00 2001
From: tsinggggg <c29qian@uwaterloo.ca>
Date: Sun, 9 Feb 2025 04:29:27 -0500
Subject: [PATCH] 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 <cheng.qian@ibm.com>
Co-authored-by: Cheng Qian <cheng.qian@ibm.com>
---
 src/unitxt/inference.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/unitxt/inference.py b/src/unitxt/inference.py
index 381e4edeec..c9c404e626 100644
--- 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: