SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix env. variable type issue in testing (#21609)

Browse Source 
* fix env issue

* fix env issue

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-02-13 20:53:26 +01:00
committed by GitHub
parent 5987e0ab69
commit cbecf121cd
5 changed files with 13 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/wav2vec2/test_modeling_tf_wav2vec2.py
View File

@@ -19,7 +19,6 @@ import glob
import inspect
import math
import multiprocessing
import os
import traceback
import unittest
@@ -682,7 +681,4 @@ class TFWav2Vec2ModelIntegrationTest(unittest.TestCase):
@require_pyctcdecode
@require_librosa
def test_wav2vec2_with_lm_invalid_pool(self):
timeout = os.environ.get("PYTEST_TIMEOUT", 600)
run_test_in_subprocess(
test_case=self, target_func=_test_wav2vec2_with_lm_invalid_pool, inputs=None, timeout=timeout
)
run_test_in_subprocess(test_case=self, target_func=_test_wav2vec2_with_lm_invalid_pool, inputs=None)