20 lines
776 B
Python
20 lines
776 B
Python
|
from plugins.plugins import BasePlugin
|
||
|
|
import logging
|
||
|
|
from colorama import Fore, Style
|
||
|
|
|
||
|
|
class ExampleLoggingPlugin(BasePlugin):
|
||
|
|
def __init__(self):
|
||
|
|
print(f"{Fore.LIGHTBLUE_EX}ExampleLoggingPlugin init{Style.RESET_ALL}")
|
||
|
|
# Setup any state variables here
|
||
|
|
pass
|
||
|
|
|
||
|
|
def on_epoch_start(self, **kwargs):
|
||
|
|
logging.info(f"{Fore.LIGHTBLUE_EX} ** ExampleLoggingPlugin: on_epoch_start{Style.RESET_ALL}")
|
||
|
|
for k, v in kwargs.items():
|
||
|
|
logging.info(f" {Fore.BLUE}{k}: {v}{Style.RESET_ALL}")
|
||
|
|
|
||
|
|
def on_epoch_end(self, **kwargs):
|
||
|
|
logging.info(f"{Fore.LIGHTBLUE_EX} ** ExampleLoggingPlugin: on_epoch_end{Style.RESET_ALL}")
|
||
|
|
for k, v in kwargs.items():
|
||
|
|
logging.info(f" {Fore.BLUE}{k}: {v}{Style.RESET_ALL}")
|