small inheritance fix

SENERGY-Platform · May 28, 2024 · 6ce7a49 · 6ce7a49
1 parent 653b8fe
commit 6ce7a49
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/algo/curve_anomaly/cont_device/offline_detector.py b/algo/curve_anomaly/cont_device/offline_detector.py
@@ -1,12 +1,12 @@
-from algo.curve_anomaly.cont_device.cont_detector import CurveDetector
+from algo.curve_anomaly.cont_device.cont_detector import ContCurveDetector
 from algo import utils
 from .. import cont_device
 
 import pandas as pd 
 import torch 
 
 
-class OfflineTrainContCurveDetector(CurveDetector):
+class OfflineTrainContCurveDetector(ContCurveDetector):
     # Used for training inside the operator
     def __init__(self, data_path, device_type, init_median, first_data_time):
         super().__init__(data_path, device_type, init_median, first_data_time)

diff --git a/algo/curve_anomaly/cont_device/online_detector.py b/algo/curve_anomaly/cont_device/online_detector.py
@@ -1,6 +1,6 @@
-from algo.curve_anomaly.cont_device.cont_detector import CurveDetector
+from algo.curve_anomaly.cont_device.cont_detector import ContCurveDetector
 
-class OnlineTrainContCurveDetector(CurveDetector):
+class OnlineTrainContCurveDetector(ContCurveDetector):
     # Used for outsourced training of the model 
     def __init__(self, data_path, device_type, init_median, first_data_time):
         super().__init__(data_path, device_type, init_median, first_data_time)