From e696dbda57df9869d56e9da3f140acab5370168f Mon Sep 17 00:00:00 2001 From: Marvin Ritter Date: Fri, 21 Jul 2023 04:17:36 -0700 Subject: [PATCH] Create separate module for TF-compatible `parameter_overview`. PiperOrigin-RevId: 549908101 --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index cec77ee..e94ba8c 100644 --- a/setup.py +++ b/setup.py @@ -33,7 +33,7 @@ setup( name="clu", version="0.0.9", - description=("Set of libraries for ML training loops in JAX."), + description="Set of libraries for ML training loops in JAX.", author="Common Loop Utils Authors", author_email="no-reply@google.com", long_description=open("README.md").read(), @@ -49,7 +49,7 @@ "jax", "jaxlib", "ml_collections", - "numpy", + "numpy==1.23.1", "packaging", "typing_extensions", "wrapt",