refactor: use FormulaPoisonPillMessageHandler to handle PoisonPillMes…

…sage
powerapi-ng · Jul 15, 2019 · de06bf2 · de06bf2
1 parent 69ff345
commit de06bf2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rapl_formula/rapl_formula_actor.py b/rapl_formula/rapl_formula_actor.py
@@ -34,7 +34,7 @@
 from functools import reduce
 
 from powerapi.actor import Actor, SocketInterface
-from powerapi.formula import FormulaActor, FormulaState
+from powerapi.formula import FormulaActor, FormulaState, FormulaPoisonPillMessageHandler
 from powerapi.handler import PoisonPillMessageHandler
 from powerapi.message import PoisonPillMessage
 from powerapi.report import HWPCReport
@@ -81,5 +81,5 @@ def setup(self):
         Setup the RAPL formula.
         """
         FormulaActor.setup(self)
-        self.add_handler(PoisonPillMessage, PoisonPillMessageHandler(self.state))
+        self.add_handler(PoisonPillMessage, FormulaPoisonPillMessageHandler(self.state))
         self.add_handler(HWPCReport, RAPLHandler(self.state))