Skip to content

Commit b478287

Browse files
committed
use lowercase in InfoType enum
1 parent 41e56ec commit b478287

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

src/data_designer/config/utils/info.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@
1212

1313

1414
class InfoType(str, Enum):
15-
SAMPLERS = "SAMPLERS"
16-
MODEL_CONFIGS = "MODEL_CONFIGS"
17-
MODEL_PROVIDERS = "MODEL_PROVIDERS"
15+
SAMPLERS = "samplers"
16+
MODEL_CONFIGS = "model_configs"
17+
MODEL_PROVIDERS = "model_providers"
1818

1919

2020
ConfigBuilderInfoType = Literal[InfoType.SAMPLERS, InfoType.MODEL_CONFIGS]

tests/config/utils/test_info.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def test_config_builder_unsupported_info_type(stub_model_configs):
3838
info = ConfigBuilderInfo(model_configs=stub_model_configs)
3939
with pytest.raises(
4040
ValueError,
41-
match="Unsupported info_type: 'unsupported_type'. ConfigBuilderInfo only supports 'SAMPLERS' and 'MODEL_CONFIGS'.",
41+
match="Unsupported info_type: 'unsupported_type'. ConfigBuilderInfo only supports 'samplers' and 'model_configs'.",
4242
):
4343
info.display("unsupported_type")
4444

@@ -53,6 +53,6 @@ def test_interface_model_providers_info(mock_display_model_providers_table, stub
5353
def test_interface_unsupported_info_type(stub_model_providers):
5454
info = InterfaceInfo(model_providers=stub_model_providers)
5555
with pytest.raises(
56-
ValueError, match="Unsupported info_type: 'unsupported_type'. InterfaceInfo only supports 'MODEL_PROVIDERS'."
56+
ValueError, match="Unsupported info_type: 'unsupported_type'. InterfaceInfo only supports 'model_providers'."
5757
):
5858
info.display("unsupported_type")

0 commit comments

Comments
 (0)