fix(test_async_fn.py): handle hf tgi model not being loaded yet

2024-02-15 18:38:03 -08:00 · 2024-02-15 18:38:03 -08:00 · 31c8d62ac2
commit 31c8d62ac2
parent 74381a2d06
1 changed files with 2 additions and 0 deletions
--- a/litellm/tests/test_async_fn.py
+++ b/litellm/tests/test_async_fn.py
@ -192,6 +192,8 @@ async def test_hf_completion_tgi():
        )
        # Add any assertions here to check the response
        print(response)
+    except litellm.APIError as e:
+        pass
    except litellm.Timeout as e:
        pass
    except Exception as e: