adding to test_plugin.py; renaming finalization

grid-parity-exchange · Aug 18, 2021 · aeb4017 · aeb4017
1 parent 159f02a
commit aeb4017
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/prescient/plugins/internal.py b/prescient/plugins/internal.py
@@ -26,7 +26,7 @@ def __init__(self):
                      'before_operations_solve',
                      'before_ruc_solve',
                      'after_operations',
-                     'after_simulation']
+                     'finalization']
         for cb in callbacks:
             self._setup_callback(cb)
 

diff --git a/prescient/plugins/plugin_registration.py b/prescient/plugins/plugin_registration.py
@@ -104,10 +104,10 @@ def register_update_operations_stats_callback(callback: Callable[[Options, Simul
     '''
     get_active_plugin_manager().register_update_operations_stats_callback(callback)
 
-def register_after_simulation_callback(callback: Callable[[Options, Simulator], None]) -> None:
+def register_finalization_callback(callback: Callable[[Options, Simulator], None]) -> None:
     ''' Request a method be called after prescient simulation is over.
     '''
-    get_active_plugin_manager().register_after_simulation_callback(callback)
+    get_active_plugin_manager().register_finalization_callback(callback)
 
 
 

diff --git a/prescient/simulator/simulator.py b/prescient/simulator/simulator.py
@@ -105,7 +105,7 @@ def simulate(self, options):
 
         stats_manager.end_simulation()
 
-        self.plugin_manager.invoke_after_simulation_callbacks(options, self)
+        self.plugin_manager.invoke_finalization_callbacks(options, self)
 
         print("Simulation Complete")
         import time

diff --git a/tests/simulator_tests/test_cases/test_plugin.py b/tests/simulator_tests/test_cases/test_plugin.py
@@ -55,3 +55,7 @@ def after_operations_callback(options, simulator, sced_model):
 def update_operations_stats_callback(options, simulator, operations_stats):
     msg('update_operations_stats_callback', options)
 pplugins.register_update_operations_stats_callback(update_operations_stats_callback)
+
+def finalization_callback(options, simulator):
+    msg('finalization_callback', options)
+pplugins.register_finalization_callback(finalization_callback)