type error in ML workflow

CrayLabs · Aug 2, 2024 · f9c9d56 · f9c9d56
1 parent 202f629
commit f9c9d56
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/smartsim/_core/generation/generator.py b/smartsim/_core/generation/generator.py
@@ -154,7 +154,7 @@ def _build_operations(self, job: Job, job_path: pathlib.Path) -> None:
         self._write_tagged_files(app, job_path)
 
     @staticmethod
-    def _copy_files(files: EntityFiles | None, dest: pathlib.Path) -> None:
+    def _copy_files(files: t.Union[EntityFiles, None], dest: pathlib.Path) -> None:
         """Perform copy file sys operations on a list of files.
 
         :param app: The Application attached to the Job
@@ -189,7 +189,7 @@ def _copy_files(files: EntityFiles | None, dest: pathlib.Path) -> None:
                 )
 
     @staticmethod
-    def _symlink_files(files: EntityFiles | None, dest: pathlib.Path) -> None:
+    def _symlink_files(files: t.Union[EntityFiles, None], dest: pathlib.Path) -> None:
         """Perform symlink file sys operations on a list of files.
 
         :param app: The Application attached to the Job