Unverified Commit ea838014 authored by Rush Tehrani's avatar Rush Tehrani Committed by GitHub
Browse files

fix: Fix ruamel import (#3430)

parent 650a39b7
...@@ -6,7 +6,7 @@ import dataclasses ...@@ -6,7 +6,7 @@ import dataclasses
from pathlib import Path from pathlib import Path
from typing import Any, Dict, Optional, Type, TypeVar from typing import Any, Dict, Optional, Type, TypeVar
from ruamel import yaml import ruamel.yaml as yaml
from . import util from . import util
...@@ -70,7 +70,7 @@ class ConfigBase: ...@@ -70,7 +70,7 @@ class ConfigBase:
Load config from YAML (or JSON) file. Load config from YAML (or JSON) file.
Keys in YAML file can either be camelCase or snake_case. Keys in YAML file can either be camelCase or snake_case.
""" """
data = yaml.safe_load(open(path)) data = yaml.load(open(path), Loader=yaml.SafeLoader)
if not isinstance(data, dict): if not isinstance(data, dict):
raise ValueError(f'Content of config file {path} is not a dict/object') raise ValueError(f'Content of config file {path} is not a dict/object')
return cls(**data, _base_path=Path(path).parent) return cls(**data, _base_path=Path(path).parent)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment