works with pandas < 2 and pandas 2

SimonBlanke · Jun 28, 2023 · f864363 · f864363
1 parent 4bfacd4
commit f864363
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -34,6 +34,7 @@ jobs:
           python -m pip install --upgrade pip
           python -m pip install surfaces flake8 pytest
           python -m pip install .
+          python -m pip install --upgrade pandas # surfaces install pandas < 2
 
       - name: Lint with flake8
         run: |

diff --git a/gradient_free_optimizers/optimizers/smb_opt/smbo.py b/gradient_free_optimizers/optimizers/smb_opt/smbo.py
@@ -34,7 +34,7 @@ def init_warm_start_smbo(self, search_data):
         if search_data is not None:
             # filter out nan and inf
             warm_start_smbo = search_data[
-                ~search_data.isin([np.nan, np.inf, -np.inf]).any(1)
+                ~search_data.isin([np.nan, np.inf, -np.inf]).any(axis=1)
             ]
 
             # filter out elements that are not in search space

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 scipy <2.0.0
 numpy >=1.18.1, <2.0.0
-pandas <2.0.0
+pandas
 scikit-learn >=0.21, !=0.23.*
 tqdm >=4.48.0, <5.0.0