Format

Author: hvdbm

cluv/cli/init.py

@@ -195,11 +195,15 @@ def check_ssh_hostnames(clusters: list[str]) -> None:
     missing_clusters = set(clusters).difference(ssh_hostnames)
 
     if len(missing_clusters) > 0:
-        console.print(f"[yellow]⚠️  Warning: Missing SSH config for {len(missing_clusters)} clusters. Try to run [bold]mila init[/bold] to add all available clusters.[/yellow]")
+        console.print(
+            f"[yellow]⚠️  Warning: Missing SSH config for {len(missing_clusters)} clusters. Try to run [bold]mila init[/bold] to add all available clusters.[/yellow]"
+        )
         for cluster in missing_clusters:
             console.print(f"[yellow]    - {cluster}[/yellow]")
     else:
-        console.print("[green]✅ All clusters in the cluv config are present in your SSH config.[/green]")
+        console.print(
+            "[green]✅ All clusters in the cluv config are present in your SSH config.[/green]"
+        )
 
 
 def check_job_script(project_root: Path, results_path: str) -> None:
@@ -224,7 +228,9 @@ def check_job_script(project_root: Path, results_path: str) -> None:
     for script_template in script_templates:
         script_path = scripts_dir / script_template.name
         if script_path.exists():
-            console.print(f"[green]✅ Job template script already exists at '{script_path}'.[/green]")
+            console.print(
+                f"[green]✅ Job template script already exists at '{script_path}'.[/green]"
+            )
             continue
         script_content = script_template.read_text()
         script_content = re.sub(
@@ -286,7 +292,9 @@ def _get_script_templates_path() -> Path:
         if script_templates_path.exists():
             return script_templates_path
     checked_paths_text = ", ".join(str(path) for path in checked_paths)
-    raise RuntimeError(f"Couldn't find the script templates folder. Checked: {checked_paths_text}.")
+    raise RuntimeError(
+        f"Couldn't find the script templates folder. Checked: {checked_paths_text}."
+    )
 
 
 def _get_pyproject_template_path() -> Path:

tests/test_init.py

@@ -65,13 +65,15 @@ def test_add_missing_cluv_config(self, tmp_path: Path) -> None:
     def test_keep_existing_cluv_config(self, tmp_path: Path) -> None:
         """check_cluv_config() should not overwrite an existing cluv config"""
         p = tmp_path / "pyproject.toml"
-        p.write_text(textwrap.dedent(
-            """\
+        p.write_text(
+            textwrap.dedent(
+                """\
             [tool.cluv]
             clusters = {"mila" = {}}
             results_path = "results"
             """
-        ))
+            )
+        )
 
         check_cluv_config(p)
         config = load_cluv_config(p)
@@ -193,7 +195,7 @@ def test_replace_results_dir_from_legacy_template(
         generated_legacy_script = project_root / "scripts" / "legacy_job.sh"
         assert generated_legacy_script.exists()
         generated_legacy_script_content = generated_legacy_script.read_text()
-        assert '#SBATCH --output=outputs/%j/slurm-%j.out' in generated_legacy_script_content
+        assert "#SBATCH --output=outputs/%j/slurm-%j.out" in generated_legacy_script_content
         assert 'results_path="outputs"' in generated_legacy_script_content
         assert "Using $results_path" in generated_legacy_script_content
         assert "results_dir" not in generated_legacy_script_content