diff --git a/src/cloudai/workloads/ai_dynamo/ai_dynamo.py b/src/cloudai/workloads/ai_dynamo/ai_dynamo.py index 1c92ff437..9bd558fe7 100644 --- a/src/cloudai/workloads/ai_dynamo/ai_dynamo.py +++ b/src/cloudai/workloads/ai_dynamo/ai_dynamo.py @@ -510,6 +510,7 @@ def installables(self) -> list[Installable]: self.script, self.hf_model, *self.cmd_args.installables, + *super().installables, ] if self.dcgm_exporter_image: installables.append(self.dcgm_exporter_image) diff --git a/tests/workloads/ai_dynamo/test_command_gen_strategy_slurm.py b/tests/workloads/ai_dynamo/test_command_gen_strategy_slurm.py index d3f7bb1a1..388420964 100644 --- a/tests/workloads/ai_dynamo/test_command_gen_strategy_slurm.py +++ b/tests/workloads/ai_dynamo/test_command_gen_strategy_slurm.py @@ -130,6 +130,19 @@ def test_container_mounts(strategy: AIDynamoSlurmCommandGenStrategy, test_run: T assert mounts == expected +def test_installables_include_top_level_git_repos(cmd_args: AIDynamoCmdArgs) -> None: + repo = GitRepo(url="https://github.com/example/custom-tools.git", commit="main") + tdef = AIDynamoTestDefinition( + name="test", + description="desc", + test_template_name="template", + cmd_args=cmd_args, + git_repos=[repo], + ) + + assert repo in tdef.installables + + @pytest.mark.parametrize( "module, config, expected", [