From 4ac8de4ecb829e3e51763730e30b677d43cef1a0 Mon Sep 17 00:00:00 2001 From: daniel Date: Fri, 1 Dec 2023 09:32:31 +0100 Subject: [PATCH] refactor: Make K8sPreprocessorActor generic regarding type of managed reports --- src/powerapi/processor/pre/k8s/k8s_pre_processor_actor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/powerapi/processor/pre/k8s/k8s_pre_processor_actor.py b/src/powerapi/processor/pre/k8s/k8s_pre_processor_actor.py index 841907fc..c5baecb4 100644 --- a/src/powerapi/processor/pre/k8s/k8s_pre_processor_actor.py +++ b/src/powerapi/processor/pre/k8s/k8s_pre_processor_actor.py @@ -42,7 +42,7 @@ from powerapi.processor.pre.k8s.k8s_pre_processor_handlers import K8sPreProcessorActorHWPCReportHandler, \ K8sPreProcessorActorStartMessageHandler, K8sPreProcessorActorPoisonPillMessageHandler from powerapi.processor.processor_actor import ProcessorState, ProcessorActor -from powerapi.report import HWPCReport +from powerapi.report.report import Report ADDED_EVENT = 'ADDED' DELETED_EVENT = 'DELETED' @@ -211,6 +211,6 @@ def setup(self): """ ProcessorActor.setup(self) self.add_handler(message_type=StartMessage, handler=K8sPreProcessorActorStartMessageHandler(state=self.state)) - self.add_handler(message_type=HWPCReport, handler=K8sPreProcessorActorHWPCReportHandler(state=self.state)) + self.add_handler(message_type=Report, handler=K8sPreProcessorActorHWPCReportHandler(state=self.state)) self.add_handler(message_type=PoisonPillMessage, handler=K8sPreProcessorActorPoisonPillMessageHandler(state=self.state))