diff --git a/models/ModelBase.py b/models/ModelBase.py index eed768a..75cd4fe 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -494,14 +494,12 @@ class ModelBase(object): try: with open(fun(), 'r') as file, open(models.get_config_schema_path(), 'r') as schema: - - data = yaml.safe_load(file) validate(data, yaml.safe_load(schema)) except FileNotFoundError: return {} except ValidationError as ve: - io.log_err("%s"%ve) + io.log_err(f"{ve}") return None for key, value in data.items(): diff --git a/requirements-colab.txt b/requirements-colab.txt index 9d638f5..a4b4399 100644 --- a/requirements-colab.txt +++ b/requirements-colab.txt @@ -11,3 +11,4 @@ tensorflow-gpu==2.4.0 tf2onnx==1.9.3 tensorboardX crc32c +jsonschema diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 1cc4902..ce516c7 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -14,3 +14,4 @@ Flask==1.1.1 flask-socketio==4.2.1 tensorboardX crc32c +jsonschema