From d7c5e7e9f13ecefa62f42e8dcaae90d7b28e78b8 Mon Sep 17 00:00:00 2001
From: Cioscos <claudiocicca1998@gmail.com>
Date: Sun, 5 Dec 2021 12:21:34 +0100
Subject: [PATCH] Minor update

- Updated requirements
- Little refactoring in models/ModelBase.py
---
 models/ModelBase.py    | 4 +---
 requirements-colab.txt | 1 +
 requirements-cuda.txt  | 1 +
 3 files changed, 3 insertions(+), 3 deletions(-)

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