[Breaking] Switch text loading to multi-threaded PyArrow loading (hug…

…gingface#548) * Switch text loading to multi-threaded PyArrow loading * updating script * try to pin some dependencies for our benchmarks * run benchmarks in venv * test * remove venv in benchmarks... * try using python3 * revert pip install changes for dvc benchmarks * go back to python instead of python3 Co-authored-by: Quentin Lhoest <[email protected]>
sywangyi · Sep 8, 2020 · 6f74779 · 6f74779
1 parent b74eed3
commit 6f74779
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -73,9 +73,10 @@
 ]
 
 BENCHMARKS_REQUIRE = [
-    'tensorflow',
-    'torch',
-    'transformers',
+    'numpy==1.18.5',
+    'tensorflow==2.3.0',
+    'torch==1.6.0',
+    'transformers==3.0.2',
 ]
 
 TESTS_REQUIRE = [