From c3eda9ad3e3fa250f6bd2d5d7f2c55a35bc28486 Mon Sep 17 00:00:00 2001
From: Benjamin Fineran <bfineran@users.noreply.github.com>
Date: Wed, 5 Jul 2023 15:11:42 -0400
Subject: [PATCH] [cherry-pick 1.5.3] pin pydantic to <2.0 (#1645) (#1651)

* pin pydantic to <2.0 (#1645)

* bump version to 1.5.3
---
 setup.py                | 2 +-
 src/sparseml/version.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 4e3d6149d91..75008d59442 100644
--- a/setup.py
+++ b/setup.py
@@ -43,7 +43,7 @@
     "pandas>=0.25.0,<2.0",
     "packaging>=20.0",
     "psutil>=5.0.0",
-    "pydantic>=1.5.0",
+    "pydantic>=1.8.2,<2.0.0",
     "requests>=2.0.0",
     "scikit-image>=0.15.0",
     "scikit-learn>=0.24.2",
diff --git a/src/sparseml/version.py b/src/sparseml/version.py
index 0d31af6e3f3..54680d62382 100644
--- a/src/sparseml/version.py
+++ b/src/sparseml/version.py
@@ -19,7 +19,7 @@
 from datetime import date
 
 
-version_base = "1.5.2"
+version_base = "1.5.3"
 is_release = False  # change to True to set the generated version as a release version