mirror of
https://github.com/NVIDIA-NeMo/DataDesigner
synced 2026-05-24 09:48:29 +00:00
b94b88b7a4
* feat(cli): bootstrap default configs on command run * fix(cli): use active interpreter in bootstrap warning * refactor(cli): simplify bootstrap warning flow * refactor(cli): bootstrap defaults in main entrypoint * refactor(cli): keep bootstrap ownership in main * test(cli): cover lazy dispatch and runtime failure flag * refactor(cli): remove redundant bootstrap state * test(cli): assert bootstrap warning includes error * test: address cli bootstrap review feedback
41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
import data_designer.cli.runtime as runtime_mod
|
|
|
|
|
|
def test_ensure_cli_default_model_settings_attempts_default_setup() -> None:
|
|
"""CLI bootstrap delegates to default setup when the CLI starts."""
|
|
with (
|
|
patch("data_designer.cli.runtime.print_warning") as mock_print_warning,
|
|
patch("data_designer.cli.runtime.resolve_seed_default_model_settings") as mock_resolve,
|
|
):
|
|
runtime_mod.ensure_cli_default_model_settings()
|
|
|
|
mock_resolve.assert_called_once_with()
|
|
mock_print_warning.assert_not_called()
|
|
|
|
|
|
def test_ensure_cli_default_model_settings_warns_and_continues() -> None:
|
|
"""CLI bootstrap prints an actionable warning when setup fails."""
|
|
with (
|
|
patch("data_designer.cli.runtime.print_warning") as mock_print_warning,
|
|
patch(
|
|
"data_designer.cli.runtime.resolve_seed_default_model_settings",
|
|
side_effect=RuntimeError("boom"),
|
|
) as mock_resolve,
|
|
):
|
|
runtime_mod.ensure_cli_default_model_settings()
|
|
|
|
mock_resolve.assert_called_once_with()
|
|
mock_print_warning.assert_called_once()
|
|
warning = mock_print_warning.call_args[0][0]
|
|
assert "Could not initialize default model providers and model configs automatically." in warning
|
|
assert "The command will continue." in warning
|
|
assert "boom" in warning
|
|
assert "data-designer config providers" in warning
|
|
assert "data-designer config models" in warning
|