diff --git a/poetry.lock b/poetry.lock index 1aad3f8..f3d1a0c 100644 --- a/poetry.lock +++ b/poetry.lock @@ -1,4 +1,4 @@ -# This file is automatically @generated by Poetry 2.1.2 and should not be changed by hand. +# This file is automatically @generated by Poetry 2.3.2 and should not be changed by hand. [[package]] name = "colorama" @@ -667,4 +667,4 @@ files = [ [metadata] lock-version = "2.1" python-versions = ">=3.11" -content-hash = "ea27558d9efee082aab062f0b0a8a1049489a7e78aff5f4b5a2186343c64c64a" +content-hash = "2f2c048ee31cd1433299fceb2f6104ec8e96f9135d2bb057040208619cbb2571" diff --git a/pyproject.toml b/pyproject.toml index 68d1cb1..fa450ba 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -35,7 +35,7 @@ include = [ ] [tool.poetry.dependencies] -pandas = "^2.2.2" +pandas = ">=2.2.2,<4.0.0" numpy = "^2.0.0" scipy = "^1.13.0" scikit-learn = "^1.5.0" diff --git a/tests/test_scoring.py b/tests/test_scoring.py index ded7b01..90a92fa 100644 --- a/tests/test_scoring.py +++ b/tests/test_scoring.py @@ -297,7 +297,7 @@ def test_numerai_corr_target_pow15_option(self): corr_wo_pow = numerai_corr(df[["prediction"]], df["target"], target_pow15=False) # we would expect the correlation to be higher when using the pow15 transformation # since the predictions are rank-gauss-pow1.5 transformed in numerai_corr - assert abs(corr_w_pow[0]) > abs(corr_wo_pow[0]) + assert abs(corr_w_pow.iloc[0]) > abs(corr_wo_pow.iloc[0]) def test_filter_top_bottom(self): self.assertRaises(