pandas context

thomaswmorris · Jul 19, 2023 · 82db973 · 82db973
1 parent 18e3b3f
commit 82db973
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/bloptools/bayesian/__init__.py b/bloptools/bayesian/__init__.py
@@ -589,11 +589,12 @@ def targets(self):
 
     @property
     def feasible_for_all_tasks(self):
-        with pd.option_context("mode.use_inf_as_null", True):
-            feasible = ~self.targets.isna().any(axis=1)
-            for task in self.tasks:
-                if task.min is not None:
-                    feasible &= self.targets.loc[:, task.name].values > task.transform(task.min)
+        # TODO: make this more robust
+        # with pd.option_context("mode.use_inf_as_null", True):
+        feasible = ~self.targets.isna().any(axis=1)
+        for task in self.tasks:
+            if task.min is not None:
+                feasible &= self.targets.loc[:, task.name].values > task.transform(task.min)
         return feasible
 
     # @property