From 1785615adb0adda067c9625be136e25a31da4c2b Mon Sep 17 00:00:00 2001 From: Roman Date: Mon, 9 Sep 2024 10:53:48 -0700 Subject: [PATCH] add logging type to the logging machine class --- bittensor/btlogging/loggingmachine.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/bittensor/btlogging/loggingmachine.py b/bittensor/btlogging/loggingmachine.py index b36019a6a0..ef58cecdfd 100644 --- a/bittensor/btlogging/loggingmachine.py +++ b/bittensor/btlogging/loggingmachine.py @@ -29,6 +29,7 @@ import os import sys from logging.handlers import QueueHandler, QueueListener, RotatingFileHandler +from logging import Logger from typing import NamedTuple from statemachine import State, StateMachine @@ -55,7 +56,7 @@ class LoggingConfig(NamedTuple): logging_dir: str -class LoggingMachine(StateMachine): +class LoggingMachine(StateMachine, Logger): """Handles logger states for bittensor and 3rd party libraries.""" Default = State(initial=True)