Further fixing on test files and the files being tested.

resources/src/ai/trainer.py

@@ -29,9 +29,12 @@
 import configparser
 from datetime import datetime
 from src.ai.outliers import Autoencoder
-from tensorflow import keras
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
+"""
+This module extends the Autoencoder class to allow further training of the model.
+"""
+
 class Trainer(Autoencoder):
     """
     Class for training the autoencoder model for anomaly detection.
@@ -138,4 +141,4 @@ def train(self, raw_data, epochs=20, batch_size=32, backup_path=None):
         loss = self.model_loss(prep_data, self.model.predict(prep_data), single_value=False).numpy()
         self.AVG_LOSS = 0.9*self.AVG_LOSS + 0.1*loss.mean()
         self.STD_LOSS = 0.9*self.AVG_LOSS + 0.1*loss.std()
-        self.save_model(self.model_file ,self.model_config_file)
+        self.save_model(self.model_file ,self.model_config_file)

resources/src/logger/logger.py

@@ -35,12 +35,12 @@ class Logger:
     def __init__(self, log_level=logging.INFO, log_file=None):  # Accept log_file as an argument with a default value
         if log_file is None:
             log_file = './outliers.log'
-        try:
-            from config import configmanager
-            config = configmanager.ConfigManager(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "config.ini"))
-            log_file = config.get('Logger', 'log_file')
-        except Exception as e:
-            print("Could not resolve ConfigManager, default set to ./outliers.log")
+            try:
+                from src.config import configmanager
+                config = configmanager.ConfigManager(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "config.ini"))
+                log_file = config.get('Logger', 'log_file')
+            except Exception as e:
+                print("Could not resolve ConfigManager, default set to ./outliers.log")
         self.logger = PyLogrus(name="Outlierslogger")
         log_dir = os.path.dirname(log_file)
         if not os.path.isdir(log_dir):
@@ -72,4 +72,4 @@ def debug(self, message):
     def error(self, message):
         self.logger.error(message)
 
-logger = Logger()
+logger = Logger()

resources/tests/test_outliers.py

@@ -20,7 +20,6 @@
 import unittest
 import os
 import sys
-import logging
 import json
 
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))

resources/tests/test_trainer.py

@@ -32,7 +32,7 @@ class TestTrainer(unittest.TestCase):
     def setUp(self):
         self.test_backup_path = "./resources/tests/test_backups/"
         os.makedirs(self.test_backup_path, exist_ok=True)
-        self.trainer = Trainer("./resources/tests/test_model.keras", "./resources/tests/test_model_config.ini")
+        self.trainer = Trainer("./resources/tests/model_test.keras", "./resources/tests/model_test_config.ini")
         self.trainer.model_config_file = "./resources/tests/dummy_config.ini"
         self.trainer.model_file = "./resources/tests/dummy.keras"