diff --git a/openllm-core/src/openllm_core/__init__.py b/openllm-core/src/openllm_core/__init__.py index a69c727a..aea63eb5 100644 --- a/openllm-core/src/openllm_core/__init__.py +++ b/openllm-core/src/openllm_core/__init__.py @@ -1,4 +1,4 @@ -from . import exceptions as exceptions, prompts as prompts, utils as utils +from . import exceptions as exceptions, utils as utils from ._configuration import ( GenerationConfig as GenerationConfig, LLMConfig as LLMConfig, diff --git a/openllm-python/tests/configuration_test.py b/openllm-python/tests/configuration_test.py index a2c28de8..cde032fa 100644 --- a/openllm-python/tests/configuration_test.py +++ b/openllm-python/tests/configuration_test.py @@ -164,19 +164,6 @@ def test_struct_envvar_with_overwrite_provided_env(monkeypatch: pytest.MonkeyPat assert sent.field1 == 20.0 -@given(model_settings()) -def test_click_conversion(gen_settings: ModelSettings): - # currently our conversion omit Union type. - def cli_mock(**attrs: t.Any): - return attrs - - cl_ = make_llm_config('ClickConversionLLM', gen_settings) - wrapped = cl_.parse(cli_mock) - filtered = {k for k, v in cl_.__openllm_hints__.items() if t.get_origin(v) is not t.Union} - click_options_filtered = [i for i in wrapped.__click_params__ if i.name and not i.name.startswith('fake_')] - assert len(filtered) == len(click_options_filtered) - - @pytest.mark.parametrize('model_name', openllm.CONFIG_MAPPING.keys()) def test_configuration_dict_protocol(model_name: str): config = openllm.AutoConfig.for_model(model_name)