From 5a6acd9ca596c850044003a7b29b4eefdba5b974 Mon Sep 17 00:00:00 2001 From: Christian Muise Date: Tue, 10 Dec 2024 22:56:33 -0500 Subject: [PATCH] Adding the package requirements. --- setup.py | 34 ++++++++++++++++++++++------------ 1 file changed, 22 insertions(+), 12 deletions(-) diff --git a/setup.py b/setup.py index abfe4c2..5eb1150 100644 --- a/setup.py +++ b/setup.py @@ -4,20 +4,30 @@ long_description = fh.read() setup( - name='lang2pddl', - version='0.1.3', - packages=find_packages(exclude=['tests']), - description='Library to connect LLMs and planning tasks', + name="lang2pddl", + version="0.1.4", + packages=find_packages(exclude=["tests"]), + description="Library to connect LLMs and planning tasks", long_description=long_description, long_description_content_type="text/markdown", - author='Marcus Tantakoun', - author_email='mtantakoun@gmail.com', - license='MIT', - url='https://github.com/AI-Planning/l2p', + author="Marcus Tantakoun", + author_email="mtantakoun@gmail.com", + install_requires=[ + "openai", + "tiktoken", + "retry", + "pddl", + "typing_extensions", + "transformers>=4.43.1", + "torch>=2.2", + "accelerate>=0.26.0", + ], + license="MIT", + url="https://github.com/AI-Planning/l2p", classifiers=[ - 'Programming Language :: Python :: 3', - 'License :: OSI Approved :: MIT License', - 'Operating System :: OS Independent', + "Programming Language :: Python :: 3", + "License :: OSI Approved :: MIT License", + "Operating System :: OS Independent", ], - python_requires='>=3.10', + python_requires=">=3.10", )