Dev deps cnt (#1732)

* added deps on new versions of packages Signed-off-by: Tomasz Kornuta <[email protected]> * bumped version of EFF to 0.2.6, added nvidia-pypi to setup reqs Signed-off-by: Tomasz Kornuta <[email protected]> * Using setup.py style fix to fix lack of space style in setup.py Signed-off-by: Tomasz Kornuta <[email protected]> * removed graph surgeon Signed-off-by: Tomasz Kornuta <[email protected]> * pinning version of webdataset to 0.1.40 Signed-off-by: Tomasz Kornuta <[email protected]>
NVIDIA · Feb 16, 2021 · c5cd85f · c5cd85f
1 parent 770eb6f
commit c5cd85f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -11,4 +11,5 @@ omegaconf>=2.0.5
 hydra-core>=1.0.4
 transformers>=4.0.1
 sentencepiece<1.0.0
-webdataset
+webdataset==0.1.40
+nvidia-eff>=0.2.6
diff --git a/setup.py b/setup.py
@@ -232,7 +232,7 @@ def finalize_options(self):
     ],
     packages=setuptools.find_packages(),
     install_requires=install_requires,
-    setup_requires=['pytest-runner'],
+    setup_requires=['pytest-runner', 'nvidia-pyindex'],
     tests_require=tests_requirements,
     # List additional groups of dependencies here (e.g. development
     # dependencies). You can install these using the following syntax,