fix (#737)

5 years ago · 0363558b91
1 changed files with 2 additions and 1 deletions
--- 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: