/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
tests/test_config/test_load_config.py
17 строк
617 B
Hongxin Liu
[misc] update pre-commit and run all files (#4752)
19 сен 2023, 09:20
Не верифицирован
19 сен 2023, 09:20
079bf3c
Код
Авторство
О чём код?
#!/usr/bin/env python # -*- encoding: utf-8 -*- from pathlib import Path from colossalai.context.config import Config def test_load_config(): filename = Path(__file__).parent.joinpath("sample_config.py") config = Config.from_file(filename) assert config.train_data, "cannot access train data as attribute" assert config.train_data.dataset, "cannot access grandchild attribute" assert isinstance( config.train_data.dataset.transform_pipeline[0], dict ), f"expected attribute transform_pipeline elements to be a dict, but found {type(config.train_data.dataset.transform_pipeline)}"