Skip to content

Commit be4e12b

Browse files
[pre-commit.ci] Add auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 2ebaa44 commit be4e12b

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

vec_inf/cli/_cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def cli() -> None:
122122
@click.option(
123123
"--env",
124124
type=str,
125-
help="Environment variables to be set. Seperate variables with commas. Can also include path to a file containing environment variables seperated by newlines. e.g. --env 'TRITON_CACHE_DIR=/scratch/.cache/triton,my_custom_vars_file.env'"
125+
help="Environment variables to be set. Seperate variables with commas. Can also include path to a file containing environment variables seperated by newlines. e.g. --env 'TRITON_CACHE_DIR=/scratch/.cache/triton,my_custom_vars_file.env'",
126126
)
127127
def launch(
128128
model_name: str,

vec_inf/client/_slurm_script_generator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def _generate_server_setup(self) -> str:
107107
SLURM_SCRIPT_TEMPLATE["singularity_command"].format(
108108
model_weights_path=self.model_weights_path,
109109
additional_binds=self.additional_binds,
110-
env_str=self.env_str
110+
env_str=self.env_str,
111111
),
112112
)
113113
else:
@@ -140,7 +140,7 @@ def _generate_launch_cmd(self) -> str:
140140
SLURM_SCRIPT_TEMPLATE["singularity_command"].format(
141141
model_weights_path=self.model_weights_path,
142142
additional_binds=self.additional_binds,
143-
env_str=self.env_str
143+
env_str=self.env_str,
144144
)
145145
)
146146
else:

0 commit comments

Comments
 (0)