Adjust tests

Signed-off-by: Wojciech Prazuch <[email protected]>

Author: wprazuch

packages/nemo-evaluator-launcher/src/nemo_evaluator_launcher/executors/slurm/executor.py

@@ -736,7 +736,6 @@ def _open_master_connection(
     suffix: str = "",
 ) -> str | None:
     ssh_command = f"ssh -MNf -S {socket} {username}{suffix}@{hostname}"
-    print(f"SSH Command to be executed: {ssh_command}")
     completed_process = subprocess.run(
         args=shlex.split(ssh_command), capture_output=True
     )

packages/nemo-evaluator-launcher/tests/unit_tests/test_slurm_executor.py

@@ -1778,7 +1778,7 @@ def test_read_autoresumed_slurm_job_ids(self, monkeypatch):
         # Mock _read_files_from_remote to return job ID lists
         monkeypatch.setattr(
             "nemo_evaluator_launcher.executors.slurm.executor._read_files_from_remote",
-            lambda paths, user, host, sock: ["123 456 789", "111 222"],
+            lambda paths, user, host, sock, suffix="": ["123 456 789", "111 222"],
             raising=True,
         )
 
@@ -1841,7 +1841,7 @@ def test_get_progress_with_dataset_size(self, monkeypatch):
         # Mock file reads
         monkeypatch.setattr(
             "nemo_evaluator_launcher.executors.slurm.executor._read_files_from_remote",
-            lambda paths, user, host, sock: ["100", "config: test"]
+            lambda paths, user, host, sock, suffix="": ["100", "config: test"]
             if "progress" in str(paths[0])
             else ["framework_name: test\nconfig:\n  type: test"],
             raising=True,
@@ -1868,7 +1868,7 @@ def test_get_progress_no_dataset_size(self, monkeypatch):
         from nemo_evaluator_launcher.executors.slurm.executor import _get_progress
 
         # Mock file reads - return progress and valid config, but no dataset size
-        def mock_read_files(paths, user, host, sock):
+        def mock_read_files(paths, user, host, sock, suffix=""):
             if "progress" in str(paths[0]):
                 return ["150"]
             else:  # run_config paths
@@ -1903,7 +1903,7 @@ def test_get_progress_missing_files(self, monkeypatch):
         # Mock file reads to return empty strings (files not found)
         monkeypatch.setattr(
             "nemo_evaluator_launcher.executors.slurm.executor._read_files_from_remote",
-            lambda paths, user, host, sock: [""],
+            lambda paths, user, host, sock, suffix="": [""],
             raising=True,
         )