Fix tests

Author: nabinchha

src/data_designer/config/utils/constants.py

@@ -259,4 +259,4 @@ class NordColor(Enum):
 ]
 
 NVIDIA_PROVIDER_NAME = "nvidia"
-NVIDIA_API_KEY_ENV_VAR_NAME = "NVIDIA_API_KEY"
+NVIDIA_API_KEY_ENV_VAR_NAME = "NVIDIA_API_KEY1"

src/data_designer/config/utils/info.py

@@ -61,7 +61,7 @@ def display(self, info_type: ConfigBuilderInfoType, **kwargs) -> None:
             display_model_configs_table(self._model_configs)
         else:
             raise ValueError(
-                f"Unsupported info_type: {info_type!r}. "
+                f"Unsupported info_type: {str(info_type)!r}. "
                 f"ConfigBuilderInfo only supports {InfoType.SAMPLERS.value!r} and {InfoType.MODEL_CONFIGS.value!r}."
             )
 
@@ -93,5 +93,5 @@ def display(self, info_type: DataDesignerInfoType, **kwargs) -> None:
             display_model_providers_table(self._model_providers)
         else:
             raise ValueError(
-                f"Unsupported info_type: {info_type!r}. InterfaceInfo only supports {InfoType.MODEL_PROVIDERS.value!r}."
+                f"Unsupported info_type: {str(info_type)!r}. InterfaceInfo only supports {InfoType.MODEL_PROVIDERS.value!r}."
             )

tests/config/test_config_builder.py

@@ -32,6 +32,7 @@
 from data_designer.config.sampler_params import SamplerType, UUIDSamplerParams
 from data_designer.config.seed import DatastoreSeedDatasetReference, SamplingStrategy
 from data_designer.config.utils.code_lang import CodeLang
+from data_designer.config.utils.info import ConfigBuilderInfo
 from data_designer.config.validator_params import CodeValidatorParams
 
 
@@ -97,7 +98,7 @@ def test_from_config(stub_data_designer_builder_config_str, mock_fetch_seed_data
 
 def test_info(stub_data_designer_builder):
     assert stub_data_designer_builder.info is not None
-    assert isinstance(stub_data_designer_builder.info.sampler_types, list)
+    assert isinstance(stub_data_designer_builder.info, ConfigBuilderInfo)
 
 
 def test_add_column_with_types(stub_empty_builder):

tests/engine/test_secret_resolver.py

@@ -85,5 +85,5 @@ def test_composite_resolver_error(stub_secrets_file: Path):
         resolver.resolve("QUUX")
 
     # The composite error message aggregates the individual resolvers' error messages
-    assert "env var" in str(excinfo.value)
+    assert "Environment variable" in str(excinfo.value)
     assert "secret" in str(excinfo.value)