From de06bf288e5a6b7c6263ff61e1350d5b325abdde Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Arthur=20d=27Az=C3=A9mar?= Date: Mon, 15 Jul 2019 17:07:28 +0200 Subject: [PATCH] refactor: use FormulaPoisonPillMessageHandler to handle PoisonPillMessage --- rapl_formula/rapl_formula_actor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rapl_formula/rapl_formula_actor.py b/rapl_formula/rapl_formula_actor.py index 8e8277e..bc31148 100644 --- 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))