From 721b068e59d77fa733127d6c02df738f7e101983 Mon Sep 17 00:00:00 2001 From: Sebastian Peralta Date: Sat, 3 Feb 2024 22:27:18 -0500 Subject: [PATCH 1/2] Update setup.py --- setup.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 1ca73a37..bd48ce8e 100644 --- a/setup.py +++ b/setup.py @@ -1,3 +1,14 @@ from setuptools import setup -setup(name="octo", packages=["octo"]) +setup( + name="octo", + version="0.0.1", + packages=find_packages(), + python_requires=">=3.8", + install_requires=[ + "jax>=0.4.20", + "jaxlib>=0.4.20", + "dlimp @ git+https://github.com/kvablack/dlimp@d08da3852c149548aaa8551186d619d87375df08", + "tensorflow_datasets <= 4.9.3", # Required for gsutil to work as intended. See https://github.com/tensorflow/datasets/issues/5203. + ], +) From 69ac29d1503575c1af15c19137134905557bdf00 Mon Sep 17 00:00:00 2001 From: Sebastian Peralta Date: Sat, 3 Feb 2024 22:35:59 -0500 Subject: [PATCH 2/2] Update setup.py --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index bd48ce8e..16f24aea 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,4 @@ -from setuptools import setup +from setuptools import find_packages, setup setup( name="octo",