From b3128c118b421aa7a1e43a975ac24e625436de2f Mon Sep 17 00:00:00 2001 From: Ruinong Tian Date: Sat, 29 Jun 2024 03:56:01 +0000 Subject: [PATCH] remove gpu test from CPU test cases --- test/test_dockerfile_based_harness.py | 1 - 1 file changed, 1 deletion(-) diff --git a/test/test_dockerfile_based_harness.py b/test/test_dockerfile_based_harness.py index 07f6066f..f7c8dd90 100644 --- a/test/test_dockerfile_based_harness.py +++ b/test/test_dockerfile_based_harness.py @@ -45,7 +45,6 @@ ("serve.test.Dockerfile", ["serve-langchain"]), ("langchain-aws.test.Dockerfile", ["langchain-aws"]), ("mlflow.test.Dockerfile", ["mlflow"]), - ("gpu-dependencies.test.Dockerfile", ["pytorch", "tensorflow"]), ], ) def test_dockerfiles_for_cpu(