Fixed missed cast to int for learning period. in tm detector too

John74561 · Jun 13, 2017 · 7f83a6e · 7f83a6e
1 parent a0eecf1
commit 7f83a6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nab/detectors/numenta/numentaTM_detector.py b/nab/detectors/numenta/numentaTM_detector.py
@@ -67,7 +67,7 @@ def initialize(self):
     self.model.enableInference({"predictedField": "value"})
 
     # Initialize the anomaly likelihood object
-    numentaLearningPeriod = math.floor(self.probationaryPeriod / 2.0)
+    numentaLearningPeriod = int(math.floor(self.probationaryPeriod / 2.0))
     self.anomalyLikelihood = anomaly_likelihood.AnomalyLikelihood(
       claLearningPeriod=numentaLearningPeriod,
       estimationSamples=self.probationaryPeriod-numentaLearningPeriod,