diff --git a/src/text_generation/services/logging/abstract_logging_service.py b/src/text_generation/services/logging/abstract_logging_service.py index 694ba3504..4777881f6 100644 --- a/src/text_generation/services/logging/abstract_logging_service.py +++ b/src/text_generation/services/logging/abstract_logging_service.py @@ -7,4 +7,7 @@ class AbstractLoggingService(abc.ABC): logger = logging.getLogger() logger.setLevel(logging.DEBUG) logger.addHandler(handler) - self.logger = logger \ No newline at end of file + self.logger = logger + + def _get_logger(self): + return self.logger \ No newline at end of file diff --git a/src/text_generation/services/logging/file_logging_service.py b/src/text_generation/services/logging/file_logging_service.py index 06645839f..cb8a7138d 100644 --- a/src/text_generation/services/logging/file_logging_service.py +++ b/src/text_generation/services/logging/file_logging_service.py @@ -5,4 +5,4 @@ from src.text_generation.services.logging.abstract_logging_service import Abstra class FileLoggingService(AbstractLoggingService): def __init__(self, filename): super().__init__(handler=logging.FileHandler(filename)) - self.logger = super().logger \ No newline at end of file + self.logger = super()._get_logger() \ No newline at end of file