small fixes (#72)

Author: johnnygreco

src/data_designer/config/default_model_settings.py

@@ -78,7 +78,7 @@ def get_default_model_configs() -> list[ModelConfig]:
     return []
 
 
-def get_defaul_model_providers_missing_api_keys() -> list[str]:
+def get_default_model_providers_missing_api_keys() -> list[str]:
     missing_api_keys = []
     for predefined_provider in PREDEFINED_PROVIDERS:
         if os.environ.get(predefined_provider["api_key"]) is None:

src/data_designer/interface/data_designer.py

@@ -9,8 +9,8 @@
 from data_designer.config.analysis.dataset_profiler import DatasetProfilerResults
 from data_designer.config.config_builder import DataDesignerConfigBuilder
 from data_designer.config.default_model_settings import (
-    get_defaul_model_providers_missing_api_keys,
     get_default_model_configs,
+    get_default_model_providers_missing_api_keys,
     get_default_provider_name,
     get_default_providers,
     resolve_seed_default_model_settings,
@@ -313,7 +313,7 @@ def _resolve_model_providers(self, model_providers: list[ModelProvider] | None)
         if model_providers is None:
             if can_run_data_designer_locally():
                 model_providers = get_default_providers()
-                missing_api_keys = get_defaul_model_providers_missing_api_keys()
+                missing_api_keys = get_default_model_providers_missing_api_keys()
                 if len(missing_api_keys) == len(PREDEFINED_PROVIDERS):
                     logger.warning(
                         "🚨 You are trying to use a default model provider but your API keys are missing."

tests/config/test_columns.py

@@ -408,7 +408,7 @@ def test_sampler_column_config_discriminated_union_person_vs_person_from_faker()
     assert person_faker_config.params.locale == "en_GB"
 
     # Verify they are different types
-    assert type(person_config.params) != type(person_faker_config.params)
+    assert type(person_config.params) is not type(person_faker_config.params)
     assert isinstance(person_config.params, PersonSamplerParams)
     assert isinstance(person_faker_config.params, PersonFromFakerSamplerParams)
 

tests/config/test_default_model_settings.py

@@ -11,9 +11,9 @@
 from data_designer.config.default_model_settings import (
     get_builtin_model_configs,
     get_builtin_model_providers,
-    get_defaul_model_providers_missing_api_keys,
     get_default_inference_parameters,
     get_default_model_configs,
+    get_default_model_providers_missing_api_keys,
     get_default_provider_name,
     get_default_providers,
     resolve_seed_default_model_settings,
@@ -152,4 +152,4 @@ def test_resolve_seed_default_model_settings(tmp_path: Path):
 @patch("data_designer.config.default_model_settings.os.environ.get")
 def test_get_default_model_providers_missing_api_keys(mock_environ_get):
     mock_environ_get.return_value = None
-    assert get_defaul_model_providers_missing_api_keys() == ["NVIDIA_API_KEY", "OPENAI_API_KEY"]
+    assert get_default_model_providers_missing_api_keys() == ["NVIDIA_API_KEY", "OPENAI_API_KEY"]