Casts to integer inf

Author: miguelgondu

src/poli/benchmarks/guacamol.py

@@ -23,6 +23,7 @@
     Proceedings of Neural Information Processing Systems, NeurIPS Datasets and Benchmarks, 2021.
 """
 
+import math
 from typing import Literal, Union
 
 from poli.core.abstract_benchmark import AbstractBenchmark
@@ -85,7 +86,7 @@ def __init__(
         batch_size: Union[int, None] = None,
         parallelize: bool = False,
         num_workers: Union[int, None] = None,
-        evaluation_budget: int = float("inf"),
+        evaluation_budget: int = int(math.inf),
     ) -> None:
         super().__init__(
             seed=seed,

src/poli/benchmarks/pmo.py

@@ -75,7 +75,7 @@ def __init__(
         batch_size: Union[int, None] = None,
         parallelize: bool = False,
         num_workers: Union[int, None] = None,
-        evaluation_budget: int = float("inf"),
+        evaluation_budget: int = int("inf"),
     ) -> None:
         super().__init__(
             string_representation=string_representation,

src/poli/benchmarks/toy_continuous_functions_benchmark.py

@@ -54,7 +54,7 @@ def __init__(
         batch_size: Union[int, None] = None,
         parallelize: bool = False,
         num_workers: Union[int, None] = None,
-        evaluation_budget: Union[int, List[int]] = float("inf"),
+        evaluation_budget: Union[int, List[int]] = int("inf"),
     ) -> None:
         super().__init__(
             seed=seed,
@@ -121,7 +121,7 @@ def __init__(
         batch_size: Union[int, None] = None,
         parallelize: bool = False,
         num_workers: Union[int, None] = None,
-        evaluation_budget: Union[int, List[int]] = float("inf"),
+        evaluation_budget: Union[int, List[int]] = int("inf"),
     ) -> None:
         super().__init__(
             seed,
@@ -174,7 +174,7 @@ def __init__(
         batch_size: Union[int, None] = None,
         parallelize: bool = False,
         num_workers: Union[int, None] = None,
-        evaluation_budget: Union[int, List[int]] = float("inf"),
+        evaluation_budget: Union[int, List[int]] = int("inf"),
     ) -> None:
         super().__init__(
             seed,

src/poli/core/abstract_benchmark.py

@@ -16,7 +16,7 @@ def __init__(
         batch_size: Union[int, None] = None,
         parallelize: bool = False,
         num_workers: Union[int, None] = None,
-        evaluation_budget: int = float("inf"),
+        evaluation_budget: int = int("inf"),
     ) -> None:
         self.seed = seed
         self.batch_size = batch_size

src/poli/core/abstract_black_box.py

@@ -30,7 +30,7 @@ class AbstractBlackBox:
         which uses half of the available CPU cores.
     evaluation_budget : int, optional
         The maximum number of evaluations allowed for the black box function.
-        Default is float("inf").
+        Default is int("inf").
 
     Attributes
     ----------
@@ -74,7 +74,7 @@ def __init__(
         batch_size: int = None,
         parallelize: bool = False,
         num_workers: int = None,
-        evaluation_budget: int = float("inf"),
+        evaluation_budget: int = int("inf"),
         force_isolation: bool = False,
     ):
         """
@@ -89,7 +89,7 @@ def __init__(
         num_workers : int, optional
             The number of workers for parallel execution, by default we use half the available CPUs.
         evaluation_budget : int, optional
-            The maximum number of evaluations allowed for the black box function, by default float("inf").
+            The maximum number of evaluations allowed for the black box function, by default int("inf").
         """
         self.observer = None
         self.observer_info = None

src/poli/core/abstract_problem_factory.py

@@ -34,7 +34,7 @@ def create(
         batch_size: int = None,
         parallelize: bool = False,
         num_workers: int = None,
-        evaluation_budget: int = float("inf"),
+        evaluation_budget: int = int("inf"),
         force_isolation: bool = False,
     ) -> Problem:
         """

src/poli/core/chemistry/tdc_black_box.py

@@ -64,7 +64,7 @@ class TDCBlackBox(AbstractBlackBox):
 
     Methods
     -------
-    __init__(self, oracle_name, string_representation="SMILES", force_isolation=False, batch_size=None, parallelize=False, num_workers=None, evaluation_budget=float("inf"), **kwargs_for_oracle
+    __init__(self, oracle_name, string_representation="SMILES", force_isolation=False, batch_size=None, parallelize=False, num_workers=None, evaluation_budget=int("inf"), **kwargs_for_oracle
         Initializes a new instance of the abstract TDC class.
 
     References
@@ -83,7 +83,7 @@ def __init__(
         batch_size: int = None,
         parallelize: bool = False,
         num_workers: int = None,
-        evaluation_budget: int = float("inf"),
+        evaluation_budget: int = int("inf"),
         **kwargs_for_oracle,
     ):
         if parallelize:

src/poli/core/lambda_black_box.py

@@ -43,7 +43,7 @@ class LambdaBlackBox(AbstractBlackBox):
         which uses half of the available CPU cores.
     evaluation_budget : int, optional
         The maximum number of evaluations allowed for the black box function.
-        Default is float("inf").
+        Default is int("inf").
     force_isolation : bool, optional
         Flag indicating whether to force isolation of the black
         box function. In this black box, this flag is ignored.
@@ -56,7 +56,7 @@ def __init__(
         batch_size=None,
         parallelize=False,
         num_workers=None,
-        evaluation_budget=float("inf"),
+        evaluation_budget=int("inf"),
         force_isolation=False,
     ):
         super().__init__(

src/poli/core/proteins/foldx_black_box.py

@@ -89,7 +89,7 @@ def __init__(
         batch_size: int = None,
         parallelize: bool = False,
         num_workers: int = None,
-        evaluation_budget: int = float("inf"),
+        evaluation_budget: int = int("inf"),
         wildtype_pdb_path: Union[Path, List[Path]] = None,
         alphabet: List[str] = None,
         experiment_id: str = None,

src/poli/core/util/isolation/isolated_black_box.py

@@ -8,7 +8,7 @@ def __init__(
         batch_size: int = None,
         parallelize: bool = False,
         num_workers: int = None,
-        evaluation_budget: int = float("inf"),
+        evaluation_budget: int = int("inf"),
         **kwargs_for_black_box,
     ):