From 57e16fed85037f2415bac3b1a55997ac4967fd99 Mon Sep 17 00:00:00 2001 From: Lluis Gifre <lluis.gifre@cttc.es> Date: Tue, 26 Jul 2022 16:25:45 +0000 Subject: [PATCH] Fixed L3_AttackMitigator/L3_CentralizedAttackDetector - Fixed wrong imports --- src/l3_attackmitigator/service/l3_attackmitigatorService.py | 2 +- .../l3_centralizedattackdetectorServiceServicerImpl.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/l3_attackmitigator/service/l3_attackmitigatorService.py b/src/l3_attackmitigator/service/l3_attackmitigatorService.py index a359d1d9f..2ce7f9316 100644 --- a/src/l3_attackmitigator/service/l3_attackmitigatorService.py +++ b/src/l3_attackmitigator/service/l3_attackmitigatorService.py @@ -21,7 +21,7 @@ from grpc_health.v1.health_pb2_grpc import add_HealthServicer_to_server from common.proto.l3_attackmitigator_pb2_grpc import ( add_L3AttackmitigatorServicer_to_server, ) -from common.service.l3_attackmitigatorServiceServicerImpl import ( +from l3_attackmitigator.service.l3_attackmitigatorServiceServicerImpl import ( l3_attackmitigatorServiceServicerImpl, ) from l3_attackmitigator.Config import ( diff --git a/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py b/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py index ead32d248..ad05b0ee6 100644 --- a/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py +++ b/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py @@ -27,7 +27,7 @@ from common.proto.l3_centralizedattackdetector_pb2_grpc import ( ) from common.proto.l3_attackmitigator_pb2 import ( - Output, + L3AttackmitigatorOutput, ) from common.proto.l3_attackmitigator_pb2_grpc import ( L3AttackmitigatorStub, @@ -88,7 +88,7 @@ class l3_centralizedattackdetectorServiceServicerImpl(L3Centralizedattackdetecto output_message["tag_name"] = "Normal" output_message["tag"] = 0 - return Output(**output_message) + return L3AttackmitigatorOutput(**output_message) def SendInput(self, request, context): # PERFORM INFERENCE WITH SENT INPUTS -- GitLab