diff --git a/README.md b/README.md index f82c187f..d0927dff 100644 --- a/README.md +++ b/README.md @@ -37,7 +37,7 @@ or ``` git clone https://github.com/dice-group/dice-embeddings.git -conda create -n dice python=3.11 --no-default-packages && conda activate dice +conda create -n dice python=3.10 --no-default-packages && conda activate dice pip3 install "pandas>=1.5.1" pip3 install "torch>=2.0.0" pip3 install "polars>=0.16.14" diff --git a/setup.py b/setup.py index a6bbf2f9..340003ec 100644 --- a/setup.py +++ b/setup.py @@ -4,22 +4,25 @@ setup( name='dicee', description='Dice embedding is an hardware-agnostic framework for large-scale knowledge graph embedding applications', - version='0.0.3', + version='0.0.4', packages=find_packages(), - install_requires=['pandas>=1.5.1', - "polars>=0.16.14", - "pyarrow>=11.0.0", - 'torch>=2.0.0', - "pytorch-lightning==1.6.4", - "scikit-learn>=1.2.2", - "pytest>=7.2.2", - "psutil>=5.9.4", - "gradio>=3.23.0"], + install_requires=[ + "torch>=2.0.0", + "pandas>=1.5.1", + "polars>=0.16.14", + "scikit-learn>=1.2.2", + "pyarrow>=11.0.0", + "pytorch-lightning==1.6.4", + "pykeen==1.10.1", + "zstandard>=0.21.0", + "pytest>=7.2.2", + "psutil>=5.9.4", + "gradio>=3.23.0"], author='Caglar Demir', author_email='caglardemir8@gmail.com', url='https://github.com/dice-group/dice-embeddings', classifiers=[ - "Programming Language :: Python :: 3.10", + "Programming Language :: Python :: 3.11", "License :: OSI Approved :: MIT License"], python_requires='>=3.10', long_description=long_description,