From dea7423dab31cf46a4b28016d5d54433e7b49ccd Mon Sep 17 00:00:00 2001
From: Benjamin Fineran <bfineran@users.noreply.github.com>
Date: Tue, 20 Jun 2023 10:41:09 -0400
Subject: [PATCH] [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
---
 setup.py                | 2 +-
 src/sparseml/version.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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