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