[1.5.2] restrict pandas to <2.0 (#1634)

* restrict pandas to <2.0 this fixes a bug in the latest NM 1.5 supported transformers datasets that is incompatible with pandas 2.0. Future releases will support later datasets versions * bump version to 1.5.2
neuralmagic · Jun 20, 2023 · dea7423 · dea7423
1 parent 1995747
commit dea7423
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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