address #119

Author: lucidrains

pyproject.toml

@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools"]
+build-backend = "setuptools.build_meta"

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'vector_quantize_pytorch',
   packages = find_packages(),
-  version = '1.14.7',
+  version = '1.14.8',
   license='MIT',
   description = 'Vector Quantization - Pytorch',
   long_description_content_type = 'text/markdown',
@@ -16,11 +16,17 @@
     'pytorch',
     'quantization'
   ],
-  install_requires=[
-    'einops>=0.7.0',
-    'einx[torch]>=0.1.3',
+  install_requires = [
+    'einops>=0.8.0',
+    'einx[torch]>=0.2',
     'torch'
   ],
+  extras_require = dict(
+    test = [
+      'torchvision',
+      'tqdm'
+    ]
+  ),
   classifiers=[
     'Development Status :: 4 - Beta',
     'Intended Audience :: Developers',