diff --git a/tests/test_litellm/proxy/client/cli/test_pi.py b/tests/test_litellm/proxy/client/cli/test_pi.py index 3c343c9e268..6bb49566580 100644 --- a/tests/test_litellm/proxy/client/cli/test_pi.py +++ b/tests/test_litellm/proxy/client/cli/test_pi.py @@ -1,4 +1,3 @@ -import inspect import json import os import stat @@ -8,7 +7,6 @@ from pathlib import Path import pytest import requests -from litellm.proxy.client.cli.commands import pi from litellm.proxy.client.cli.commands.pi import ( ListingFailure, ModelLimits, @@ -22,11 +20,6 @@ from litellm.proxy.client.cli.commands.pi import ( ) -def test_pi_module_has_no_strenum_import(): - """enum.StrEnum is Python 3.11+; pyproject allows 3.10, so pi.py must not import it.""" - assert "StrEnum" not in inspect.getsource(pi) - - def test_listing_failure_is_str_enum(): assert issubclass(ListingFailure, str) assert ListingFailure.REJECTED.value == "rejected"