fix: correct return type annotation of get_nsys_entrypoint (NVIDIA-NeMo#568)

get_nsys_entrypoint returns a tuple (nsys, ), but its annotation was str.

Update the annotation to tuple[str, str] to match the implementation and

callers that unpack the result.

Signed-off-by: Andrew White <andrewh@cdw.com>
Signed-off-by: andrewwhitecdw <andrewwhitecdw@users.noreply.github.com>

Author: andrewwhitecdw

nemo_run/core/execution/base.py

@@ -165,7 +165,7 @@ def get_launcher_prefix(self) -> Optional[list[str]]:
             os.makedirs(os.path.join(self.job_dir, launcher.nsys_folder), exist_ok=True)
             return launcher.get_nsys_prefix(profile_dir=self.job_dir)
 
-    def get_nsys_entrypoint(self) -> str:
+    def get_nsys_entrypoint(self) -> tuple[str, str]:
         return ("nsys", "")
 
     def supports_launcher_transform(self) -> bool:

nemo_run/core/execution/slurm.py

@@ -561,7 +561,7 @@ def get_launcher_prefix(self) -> Optional[list[str]]:
                 nsys_prefix += ["$GPU_METRICS_FLAG"]
         return nsys_prefix
 
-    def get_nsys_entrypoint(self) -> str:
+    def get_nsys_entrypoint(self) -> tuple[str, str]:
         launcher = self.get_launcher()
         entrypoint, postfix = "nsys", ""
         if launcher.nsys_gpu_metrics: