From 6a400a6200b79063f74eb182e22a83ebedc5d3dc Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 10 May 2024 15:49:20 -0700 Subject: [PATCH] test: fix test --- litellm/tests/test_completion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index b3e1928a8e..04f4cc5115 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -1138,7 +1138,7 @@ def test_get_hf_task_for_model(): model = "roneneldan/TinyStories-3M" model_type = litellm.llms.huggingface_restapi.get_hf_task_for_model(model) print(f"model:{model}, model type: {model_type}") - assert model_type == None + assert model_type == "text-generation" # test_get_hf_task_for_model()