Env: dict -> Mapping

Author: MattToast

smartsim/settings/builders/launch/alps.py

@@ -220,7 +220,7 @@ def set(self, key: str, value: str | None) -> None:
         self._launch_args[key] = value
 
     def finalize(
-        self, exe: ExecutableLike, env: dict[str, str | None]
+        self, exe: ExecutableLike, env: t.Mapping[str, str | None]
     ) -> t.Sequence[str]:
         return (
             "aprun",

smartsim/settings/builders/launch/dragon.py

@@ -69,7 +69,7 @@ def set(self, key: str, value: str | None) -> None:
         self._launch_args[key] = value
 
     def finalize(
-        self, exe: ExecutableLike, env: dict[str, str | None]
+        self, exe: ExecutableLike, env: t.Mapping[str, str | None]
     ) -> DragonRunRequest:
         exe_, *args = exe.as_program_arguments()
         return DragonRunRequest(

smartsim/settings/builders/launch/local.py

@@ -79,6 +79,6 @@ def set(self, key: str, value: str | None) -> None:
         self._launch_args[key] = value
 
     def finalize(
-        self, exe: ExecutableLike, env: dict[str, str | None]
+        self, exe: ExecutableLike, env: t.Mapping[str, str | None]
     ) -> t.Sequence[str]:
         return exe.as_program_arguments()

smartsim/settings/builders/launch/lsf.py

@@ -122,7 +122,7 @@ def set(self, key: str, value: str | None) -> None:
         self._launch_args[key] = value
 
     def finalize(
-        self, exe: ExecutableLike, env: dict[str, str | None]
+        self, exe: ExecutableLike, env: t.Mapping[str, str | None]
     ) -> t.Sequence[str]:
         return (
             "jsrun",

smartsim/settings/builders/launch/mpi.py

@@ -225,7 +225,7 @@ def launcher_str(self) -> str:
         return LauncherType.Mpirun.value
 
     def finalize(
-        self, exe: ExecutableLike, env: dict[str, str | None]
+        self, exe: ExecutableLike, env: t.Mapping[str, str | None]
     ) -> t.Sequence[str]:
         return ("mpirun", *self.format_launch_args(), "--", *exe.as_program_arguments())
 
@@ -237,7 +237,7 @@ def launcher_str(self) -> str:
         return LauncherType.Mpiexec.value
 
     def finalize(
-        self, exe: ExecutableLike, env: dict[str, str | None]
+        self, exe: ExecutableLike, env: t.Mapping[str, str | None]
     ) -> t.Sequence[str]:
         return (
             "mpiexec",
@@ -254,7 +254,7 @@ def launcher_str(self) -> str:
         return LauncherType.Orterun.value
 
     def finalize(
-        self, exe: ExecutableLike, env: dict[str, str | None]
+        self, exe: ExecutableLike, env: t.Mapping[str, str | None]
     ) -> t.Sequence[str]:
         return (
             "orterun",

smartsim/settings/builders/launch/pals.py

@@ -156,7 +156,7 @@ def set(self, key: str, value: str | None) -> None:
         self._launch_args[key] = value
 
     def finalize(
-        self, exe: ExecutableLike, env: dict[str, str | None]
+        self, exe: ExecutableLike, env: t.Mapping[str, str | None]
     ) -> t.Sequence[str]:
         return (
             "mpiexec",

smartsim/settings/builders/launch/slurm.py

@@ -322,7 +322,7 @@ def set(self, key: str, value: str | None) -> None:
         self._launch_args[key] = value
 
     def finalize(
-        self, exe: ExecutableLike, env: dict[str, str | None]
+        self, exe: ExecutableLike, env: t.Mapping[str, str | None]
     ) -> t.Sequence[str]:
         return (
             "srun",

smartsim/settings/builders/launchArgBuilder.py

@@ -59,7 +59,7 @@ def set(self, arg: str, val: str | None) -> None:
         """Set the launch arguments"""
 
     @abstractmethod
-    def finalize(self, exe: ExecutableLike, env: dict[str, str | None]) -> _T:
+    def finalize(self, exe: ExecutableLike, env: t.Mapping[str, str | None]) -> _T:
         """Prepare an entity for launch using the built options"""
 
     def format_launch_args(self) -> t.Union[t.List[str], None]: