From f0602ec7494c1ab460d376d4ba9ba61bceff0e1f Mon Sep 17 00:00:00 2001 From: Michael Wyatt Date: Wed, 2 Aug 2023 11:04:19 -0700 Subject: [PATCH] unpin datasets in UT --- .github/workflows/nv-transformers-v100.yml | 2 -- 1 file changed, 2 deletions(-) diff --git a/.github/workflows/nv-transformers-v100.yml b/.github/workflows/nv-transformers-v100.yml index a8f3e216ec48..147598015ace 100644 --- a/.github/workflows/nv-transformers-v100.yml +++ b/.github/workflows/nv-transformers-v100.yml @@ -56,8 +56,6 @@ jobs: pip install .[testing] # find reqs used in ds integration tests find examples/pytorch -regextype posix-egrep -regex '.*(language-modeling|question-answering|summarization|image-classification|text-classification|translation).*/requirements.txt' -exec grep -v 'torch' {} \; | xargs -I {} pip install --upgrade {} - # force datasets version due to issues - pip install datasets==2.2.2 # force protobuf version due to issues pip install "protobuf<4.21.0" pip list