diff --git a/test-integration/test_integration/fixtures/install-requires-packaging/cog.yaml b/test-integration/test_integration/fixtures/install-requires-packaging/cog.yaml new file mode 100644 index 0000000000..bb8e89a3b4 --- /dev/null +++ b/test-integration/test_integration/fixtures/install-requires-packaging/cog.yaml @@ -0,0 +1,7 @@ +build: + python_version: "3.10" + python_requirements: requirements.txt + fast: true + gpu: true + cuda: "12.1" +predict: "predict.py:Predictor" diff --git a/test-integration/test_integration/fixtures/install-requires-packaging/predict.py b/test-integration/test_integration/fixtures/install-requires-packaging/predict.py new file mode 100644 index 0000000000..15183a07cf --- /dev/null +++ b/test-integration/test_integration/fixtures/install-requires-packaging/predict.py @@ -0,0 +1,7 @@ +from cog import BasePredictor + + +class Predictor(BasePredictor): + + def predict(self) -> str: + return "hello" diff --git a/test-integration/test_integration/fixtures/install-requires-packaging/requirements.txt b/test-integration/test_integration/fixtures/install-requires-packaging/requirements.txt new file mode 100644 index 0000000000..2c5ae3ae5b --- /dev/null +++ b/test-integration/test_integration/fixtures/install-requires-packaging/requirements.txt @@ -0,0 +1,2 @@ +causal-conv1d==1.5.0.post8 +torch==2.5.0 diff --git a/test-integration/test_integration/test_build.py b/test-integration/test_integration/test_build.py index fa4de15a98..385b89eb18 100644 --- a/test-integration/test_integration/test_build.py +++ b/test-integration/test_integration/test_build.py @@ -522,3 +522,15 @@ def test_overrides(docker_image): ) assert build_process.returncode == 0 + + +def test_install_requires_packaging(docker_image): + project_dir = Path(__file__).parent / "fixtures/install-requires-packaging" + + build_process = subprocess.run( + ["cog", "build", "-t", docker_image], + cwd=project_dir, + capture_output=True, + ) + print(build_process.stderr.decode()) + assert build_process.returncode == 0