Apply formatting fixes from pre-commit

rohan-uiuc · rohan-uiuc · commit 8b6a2119da67 · 2025-11-12T17:05:28.000-06:00
diff --git a/vec_inf/client/_slurm_script_generator.py b/vec_inf/client/_slurm_script_generator.py
@@ -121,7 +121,9 @@ def _generate_server_setup(self) -> str:
             server_script.append("\n".join(SLURM_SCRIPT_TEMPLATE["container_setup"]))
             server_script.append(
                 SLURM_SCRIPT_TEMPLATE["bind_path"].format(
-                    model_weights_path=self.model_weights_path if self.model_weights_exists else "",
+                    model_weights_path=self.model_weights_path
+                    if self.model_weights_exists
+                    else "",
                     additional_binds=self.additional_binds,
                 )
             )
@@ -289,7 +291,9 @@ def _generate_model_launch_script(self, model_name: str) -> Path:
             script_content.append(BATCH_MODEL_LAUNCH_SCRIPT_TEMPLATE["container_setup"])
         script_content.append(
             BATCH_MODEL_LAUNCH_SCRIPT_TEMPLATE["bind_path"].format(
-                model_weights_path=model_params["model_weights_path"] if model_params.get("model_weights_exists", True) else "",
+                model_weights_path=model_params["model_weights_path"]
+                if model_params.get("model_weights_exists", True)
+                else "",
                 additional_binds=model_params["additional_binds"],
             )
         )

Original file line number	Diff line number	Diff line change
`@@ -121,7 +121,9 @@ def _generate_server_setup(self) -> str:`
`121`	`121`	`server_script.append("\n".join(SLURM_SCRIPT_TEMPLATE["container_setup"]))`
`122`	`122`	`server_script.append(`
`123`	`123`	`SLURM_SCRIPT_TEMPLATE["bind_path"].format(`
`124`		`- model_weights_path=self.model_weights_path if self.model_weights_exists else "",`
	`124`	`+ model_weights_path=self.model_weights_path`
	`125`	`+ if self.model_weights_exists`
	`126`	`+ else "",`
`125`	`127`	`additional_binds=self.additional_binds,`
`126`	`128`	`)`
`127`	`129`	`)`
`@@ -289,7 +291,9 @@ def _generate_model_launch_script(self, model_name: str) -> Path:`
`289`	`291`	`script_content.append(BATCH_MODEL_LAUNCH_SCRIPT_TEMPLATE["container_setup"])`
`290`	`292`	`script_content.append(`
`291`	`293`	`BATCH_MODEL_LAUNCH_SCRIPT_TEMPLATE["bind_path"].format(`
`292`		`- model_weights_path=model_params["model_weights_path"] if model_params.get("model_weights_exists", True) else "",`
	`294`	`+ model_weights_path=model_params["model_weights_path"]`
	`295`	`+ if model_params.get("model_weights_exists", True)`
	`296`	`+ else "",`
`293`	`297`	`additional_binds=model_params["additional_binds"],`
`294`	`298`	`)`
`295`	`299`	`)`