diff --git a/setup.py b/setup.py index 8dbc147068e..4e3d6149d91 100644 --- a/setup.py +++ b/setup.py @@ -40,7 +40,7 @@ "matplotlib>=3.0.0", "merge-args>=0.1.0", "onnx>=1.5.0,<=1.12.0", - "pandas>=0.25.0", + "pandas>=0.25.0,<2.0", "packaging>=20.0", "psutil>=5.0.0", "pydantic>=1.5.0", diff --git a/src/sparseml/version.py b/src/sparseml/version.py index 21ccc54ec43..0d31af6e3f3 100644 --- a/src/sparseml/version.py +++ b/src/sparseml/version.py @@ -19,7 +19,7 @@ from datetime import date -version_base = "1.5.1" +version_base = "1.5.2" is_release = False # change to True to set the generated version as a release version