packaging updates

kingoflolz · May 21, 2021 · 4e0d6a7 · 4e0d6a7
1 parent 5490b77
commit 4e0d6a7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,6 @@ transformers~=4.4.2
 tqdm~=4.45.0
 setuptools~=51.3.3
 wandb~=0.10.22
-torch~=1.7.1
 einops~=0.3.0
 requests~=2.25.1
 fabric~=2.6.0
@@ -17,4 +16,5 @@ cloudpickle~=1.3.0
 tensorflow-cpu~=2.4.1
 google-cloud-storage~=1.36.2
 smart_open[gcs]
-func_timeout
+func_timeout
+ftfy
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
-from setuptools import setup
+from setuptools import setup, find_packages
 
 setup(
     name='mesh_transformer',
     version='0.0.0',
-    packages=['mesh_transformer']
+    packages=find_packages(include=['mesh_transformer', 'mesh_transformer.*'])
 )