diff --git a/.github/workflows/multi-gpu-e2e.yml b/.github/workflows/multi-gpu-e2e.yml index ea00d749b..4934bf9ca 100644 --- a/.github/workflows/multi-gpu-e2e.yml +++ b/.github/workflows/multi-gpu-e2e.yml @@ -4,6 +4,10 @@ on: pull_request: paths: - 'tests/e2e/multigpu/*.py' + - 'requirements.txt' + - 'setup.py' + - 'pyproject.toml' + - '.github/workflows/multi-gpu-e2e.yml' workflow_dispatch: schedule: - cron: '0 0 * * 1,4' # Runs at 00:00 UTC every monday & thursday diff --git a/cicd/multigpu.py b/cicd/multigpu.py index 2c0863034..eda589dd1 100644 --- a/cicd/multigpu.py +++ b/cicd/multigpu.py @@ -37,15 +37,11 @@ temp_dir = tempfile.mkdtemp() with open(pathlib.Path(temp_dir) / "Dockerfile", "w", encoding="utf-8") as f: f.write(dockerfile_contents) -cicd_image = ( - Image.from_dockerfile( - pathlib.Path(temp_dir) / "Dockerfile", - force_build=True, - gpu="A10G", - ) - .env(df_args) - .pip_install("fastapi==0.110.0", "pydantic==2.6.3") -) +cicd_image = Image.from_dockerfile( + pathlib.Path(temp_dir) / "Dockerfile", + force_build=True, + gpu="A10G", +).env(df_args) app = App("Axolotl CI/CD", secrets=[]) diff --git a/src/axolotl/cli/cloud/modal_.py b/src/axolotl/cli/cloud/modal_.py index 6b724f732..93d6872b1 100644 --- a/src/axolotl/cli/cloud/modal_.py +++ b/src/axolotl/cli/cloud/modal_.py @@ -123,8 +123,6 @@ class ModalCloud(Cloud): if env := self.get_env(): image = image.env(env) - image = image.pip_install("fastapi==0.110.0", "pydantic==2.6.3") - return image def get_secrets(self):