Minor update

- Updated requirements
- Little refactoring in models/ModelBase.py

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():

requirements-colab.txt

@@ -11,3 +11,4 @@ tensorflow-gpu==2.4.0
 tf2onnx==1.9.3
 tensorboardX
 crc32c
+jsonschema

requirements-cuda.txt

@@ -14,3 +14,4 @@ Flask==1.1.1
 flask-socketio==4.2.1
 tensorboardX
 crc32c
+jsonschema