Skip to content

Commit 0b02224

Browse files
committed
addressed amrits comments
1 parent f660a02 commit 0b02224

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

vec_inf/client/_helper.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -158,23 +158,24 @@ def _process_vllm_args(self, arg_string: str) -> dict[str, Any]:
158158
vllm_args[arg.strip()] = True
159159
return vllm_args
160160

161-
def _process_env_vars(self, arg_string: str) -> dict[str, str]:
161+
def _process_env_vars(self, env_arg: str) -> dict[str, str]:
162162
"""Process the env string into a dictionary of environment variables.
163163
164164
Parameters
165165
----------
166-
arg_string : str
167-
Comma separated list of environment variable definitions (eg. MY_VAR=1)
168-
or file paths containing environment variable definitions (separated by
169-
newlines)
166+
env_arg : str
167+
String containing comma separated list of environment variable definitions
168+
(eg. MY_VAR=1), file paths containing environment variable definitions
169+
(separated by newlines), or a combination of both
170+
(eg. 'MY_VAR=5,my_env.env')
170171
171172
Returns
172173
-------
173-
dict[str, str]
174-
Processed environment variables as key-value pairs.
174+
dict[str, str]
175+
Processed environment variables as key-value pairs.
175176
"""
176177
env_vars: dict[str, str] = {}
177-
for arg in arg_string.split(","):
178+
for arg in env_arg.split(","):
178179
if "=" in arg: # Arg is an env var definition
179180
key, value = arg.split("=")
180181
env_vars[key.strip()] = value.strip()

0 commit comments

Comments
 (0)