mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-02 01:28:31 +00:00
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
This commit is contained in:
@@ -3,16 +3,15 @@
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from colossalai.context.config import Config
|
||||
|
||||
|
||||
def test_load_config():
|
||||
filename = Path(__file__).parent.joinpath('sample_config.py')
|
||||
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)}'
|
||||
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)}"
|
||||
|
Reference in New Issue
Block a user