Fixed bug in ForcePredictionTarget extrapolator

chillax/chillax_extrapolator.py

@@ -268,13 +268,11 @@ class ForcePredictionTargetCHILLAXExtrapolator(CHILLAXExtrapolator):
             ic_method=ic_method,
         )
 
-        self.prediction_targets = set()
-
     def _extrapolate(self, ground_truth_uid, unconditional_probabilities):
         candidates = [
             uid
             for (uid, probability) in unconditional_probabilities.items()
-            if uid in self.prediction_targets
+            if uid in self._prediction_targets
         ]
 
         if len(candidates) > 0:

chillax/version.py

@@ -1 +1 @@
-__version__ = "0.1a4"
+__version__ = "0.1a5"