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",