Commit 97baba1b authored by Caroline Chen's avatar Caroline Chen
Browse files

Skip hubert xlarge torchscript test (#2758)

Summary:
a couple of circleci unittests are failing during hubert xlarge torchscript test, which has been known to fail on Windows in the past (#65776). this PR disables this test on circleci

cc atalman

Pull Request resolved: https://github.com/pytorch/audio/pull/2758

Reviewed By: mthrok

Differential Revision: D40290535

Pulled By: carolineechen

fbshipit-source-id: 5c5fb43434a517b6c439a8cb8e853015d1550a57
parent 6de7bb98
......@@ -184,18 +184,18 @@ class TestWav2Vec2Model(TorchaudioTestCase):
@factory_funcs
def test_pretrain_torchscript(self, factory_func):
"""Wav2Vec2Model should be scriptable"""
if factory_func is hubert_xlarge and os.name == "nt" and os.environ.get("CI") == "true":
if factory_func is hubert_xlarge and os.environ.get("CI") == "true":
self.skipTest(
"hubert_xlarge is known to fail on Windows CI. " "See https://github.com/pytorch/pytorch/issues/65776"
"hubert_xlarge is known to fail on CI. " "See https://github.com/pytorch/pytorch/issues/65776"
)
self._test_torchscript(factory_func())
@factory_funcs
def test_finetune_torchscript(self, factory_func):
"""Wav2Vec2Model should be scriptable"""
if factory_func is hubert_xlarge and os.name == "nt" and os.environ.get("CI") == "true":
if factory_func is hubert_xlarge and os.environ.get("CI") == "true":
self.skipTest(
"hubert_xlarge is known to fail on Windows CI. " "See https://github.com/pytorch/pytorch/issues/65776"
"hubert_xlarge is known to fail on CI. " "See https://github.com/pytorch/pytorch/issues/65776"
)
self._test_torchscript(factory_func(aux_num_out=32))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment