Support extra_packages in PythonPackage

ethanluoyc · Oct 5, 2023 · e68aa23 · e68aa23
1 parent 29fd8ac
commit e68aa23
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/lxm3/xm_cluster/executable_specs.py b/lxm3/xm_cluster/executable_specs.py
@@ -54,6 +54,7 @@ class PythonPackage(job_blocks.ExecutableSpec):
     entrypoint: Union[CommandList, ModuleName]
     path: str = attr.ib(converter=utils.resolve_path_relative_to_launcher, default=".")
     resources: List[Fileset] = attr.ib(converter=list, default=attr.Factory(list))
+    extra_packages: List[str] = attr.ib(converter=list, default=attr.Factory(list))
 
     @property
     def name(self) -> str:

diff --git a/lxm3/xm_cluster/packaging/__init__.py b/lxm3/xm_cluster/packaging/__init__.py
@@ -55,6 +55,7 @@ def _create_archive(
                         "-t",
                         tmpdir,
                         package_dir,
+                        *py_package.extra_packages,
                     ],
                     text=True,
                     check=True,