From f83934f5d3aea9fdd2f46151490dc9e5c618b4d9 Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Tue, 4 Jul 2023 17:29:39 -0400 Subject: [PATCH] improve plugins --- plugins/example_plugin.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 plugins/example_plugin.py diff --git a/plugins/example_plugin.py b/plugins/example_plugin.py new file mode 100644 index 0000000..afeddf5 --- /dev/null +++ b/plugins/example_plugin.py @@ -0,0 +1,19 @@ +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}")