From bc18dbd980b8b0d76ed6fa408f471ba2bcc8280b Mon Sep 17 00:00:00 2001 From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com> Date: Fri, 20 Jan 2023 15:34:11 +0100 Subject: [PATCH] skip santacoder tests --- server/tests/models/test_santacoder.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/server/tests/models/test_santacoder.py b/server/tests/models/test_santacoder.py index c3a83753..acebec04 100644 --- a/server/tests/models/test_santacoder.py +++ b/server/tests/models/test_santacoder.py @@ -42,6 +42,7 @@ def default_fim_pb_batch(default_fim_pb_request): return generate_pb2.Batch(id=0, requests=[default_fim_pb_request], size=1) +@pytest.mark.skip def test_santacoder_generate_token_completion(default_santacoder, default_pb_batch): batch = CausalLMBatch.from_pb( default_pb_batch, default_santacoder.tokenizer, default_santacoder.device @@ -65,6 +66,7 @@ def test_santacoder_generate_token_completion(default_santacoder, default_pb_bat ) +@pytest.mark.skip def test_fim_santacoder_generate_token_completion( default_santacoder, default_fim_pb_batch ):