Commit c2ea6898 authored by Caroline Chen's avatar Caroline Chen Committed by Facebook GitHub Bot
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 d735b078
...@@ -184,18 +184,18 @@ class TestWav2Vec2Model(TorchaudioTestCase): ...@@ -184,18 +184,18 @@ class TestWav2Vec2Model(TorchaudioTestCase):
@factory_funcs @factory_funcs
def test_pretrain_torchscript(self, factory_func): def test_pretrain_torchscript(self, factory_func):
"""Wav2Vec2Model should be scriptable""" """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( 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()) self._test_torchscript(factory_func())
@factory_funcs @factory_funcs
def test_finetune_torchscript(self, factory_func): def test_finetune_torchscript(self, factory_func):
"""Wav2Vec2Model should be scriptable""" """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( 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)) 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