From 7aee3ecbbaa841a2b48156d95bcb6db60436bafa Mon Sep 17 00:00:00 2001 From: Etienne Perot Date: Tue, 16 Jan 2024 16:08:50 -0800 Subject: [PATCH] Disable LLM test until our testing pipeline better supports large images. PiperOrigin-RevId: 598988194 --- Makefile | 3 ++- test/gpu/textgen_test.go | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/Makefile b/Makefile index 01765f730..205a727af 100644 --- a/Makefile +++ b/Makefile @@ -293,7 +293,8 @@ cos-gpu-smoke-tests: gpu-smoke-images $(RUNTIME_BIN) # This is a superset of those needed for smoke tests. # It includes non-GPU images that are used as part of GPU tests, # e.g. busybox and python. -gpu-images: gpu-smoke-images load-gpu_pytorch load-gpu_ollama load-basic_busybox load-basic_python +# TODO(b/306706609): Add "load-gpu_ollama" and "load-basic_busybox" back here. +gpu-images: gpu-smoke-images load-gpu_pytorch load-basic_python .PHONY: gpu-images gpu-all-tests: gpu-images gpu-smoke-tests $(RUNTIME_BIN) diff --git a/test/gpu/textgen_test.go b/test/gpu/textgen_test.go index 7e2cf35cb..a4268e5b3 100644 --- a/test/gpu/textgen_test.go +++ b/test/gpu/textgen_test.go @@ -66,6 +66,7 @@ func runSandboxedPython(ctx context.Context, logger testutil.Logger, code string // Then it asks it to write a unit test that verifies that // this text is a correct translation. func TestLLM(t *testing.T) { + t.Skip("Test is disabled until support for large images is better.") ctx := context.Background() // Run the LLM. llmContainer := dockerutil.MakeContainer(ctx, t)