mirror of
https://github.com/hanzo-ml/trainer.git
synced 2026-09-24 00:31:34 +00:00
Andrey VelichkevichandGitHub
239b27d311
* feat(controller): Refactor the Initializer APIs of TrainJob Signed-off-by: Andrey Velichkevich <[email protected]> * Fix go unit test Signed-off-by: Andrey Velichkevich <[email protected]> * Fix integration test Signed-off-by: Andrey Velichkevich <[email protected]> --------- Signed-off-by: Andrey Velichkevich <[email protected]>
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import pytest
|
|
|
|
import pkg.initializers.types.types as types
|
|
import pkg.initializers.utils.utils as utils
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"config_class,env_vars,expected",
|
|
[
|
|
(
|
|
types.HuggingFaceModelInitializer,
|
|
{"STORAGE_URI": "hf://test", "ACCESS_TOKEN": "token"},
|
|
{"storage_uri": "hf://test", "access_token": "token"},
|
|
),
|
|
(
|
|
types.HuggingFaceModelInitializer,
|
|
{"STORAGE_URI": "hf://test"},
|
|
{"storage_uri": "hf://test", "access_token": None},
|
|
),
|
|
(
|
|
types.HuggingFaceDatasetInitializer,
|
|
{"STORAGE_URI": "hf://test", "ACCESS_TOKEN": "token"},
|
|
{"storage_uri": "hf://test", "access_token": "token"},
|
|
),
|
|
(
|
|
types.HuggingFaceDatasetInitializer,
|
|
{"STORAGE_URI": "hf://test"},
|
|
{"storage_uri": "hf://test", "access_token": None},
|
|
),
|
|
],
|
|
)
|
|
def test_get_config_from_env(mock_env_vars, config_class, env_vars, expected):
|
|
mock_env_vars(**env_vars)
|
|
result = utils.get_config_from_env(config_class)
|
|
assert result == expected
|