update requirements

examples/setup_seq2seq.py

@@ -1,5 +1,5 @@
 """Install Compacter."""
-import os 
+import os
 import setuptools
 from torch.utils.cpp_extension import BuildExtension, CUDAExtension
 
@@ -29,17 +29,10 @@ def setup_package():
           'Programming Language :: Python :: 3.7.10',
       ],
       keywords='text nlp machinelearning',
-      # ext_modules=[
-      #   CUDAExtension('seq2seq.projections.fwh_cuda',
-      #       sources=[
-      #       'seq2seq/projections/fwh_cuda/fwh_cpp.cpp',
-      #       'seq2seq/projections/fwh_cuda/fwh_cu.cu',
-      #       ]
-      #   )
-      # ]
-      # ,
       cmdclass={"build_ext": BuildExtension},
       install_requires=[
+          "pyarrow==7.0.0",
+          "datasets==1.17.0"
       ],
   )
 

requirements.txt

@@ -1,6 +1,6 @@
 torch>=1.8.0
 transformers>=4.10.0
-datasets>=1.17.0
+datasets==1.17.0
 sentencepiece>=0.1.96
 tqdm>=4.62.2
 loralib