Before new release, merge branch 'develop'

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"

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='[email protected]',
     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,