diff --git a/analytics/analytics/detectors/pattern_detector.py b/analytics/analytics/detectors/pattern_detector.py index 7c0796f..936e6bc 100644 --- a/analytics/analytics/detectors/pattern_detector.py +++ b/analytics/analytics/detectors/pattern_detector.py @@ -42,7 +42,8 @@ class PatternDetector(Detector): def __init__(self, pattern_type: str, analytic_unit_id: AnalyticUnitId): super().__init__(analytic_unit_id) - self.model = resolve_model_by_pattern(pattern_type) + self.pattern_type = pattern_type + self.model = resolve_model_by_pattern(self.pattern_type) self.bucket = DataBucket() def train(self, dataframe: pd.DataFrame, segments: List[dict], cache: Optional[ModelCache]) -> ModelCache: