From 4509fe861b54070b505fbd60e0aa019a18fdb2a3 Mon Sep 17 00:00:00 2001 From: anjali411 Date: Wed, 1 Jul 2020 16:49:56 -0700 Subject: [PATCH] add import --- test/torchaudio_unittest/torchscript_consistency_cuda_test.py | 3 ++- test/torchaudio_unittest/torchscript_consistency_impl.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/test/torchaudio_unittest/torchscript_consistency_cuda_test.py b/test/torchaudio_unittest/torchscript_consistency_cuda_test.py index 50ce398e9a..c6e5b3b045 100644 --- a/test/torchaudio_unittest/torchscript_consistency_cuda_test.py +++ b/test/torchaudio_unittest/torchscript_consistency_cuda_test.py @@ -1,7 +1,7 @@ import torch from torchaudio_unittest import common_utils -from .torchscript_consistency_impl import Functional, Transforms +from .torchscript_consistency_impl import Functional, Transforms, TransformsWithComplexDtypes @common_utils.skipIfNoCuda @@ -27,6 +27,7 @@ class TestTransformsFloat64(Transforms, common_utils.PytorchTestCase): dtype = torch.float64 device = torch.device('cuda') + @common_utils.skipIfNoCuda class TestTransformsCFloat(TransformsWithComplexDtypes, common_utils.PytorchTestCase): dtype = torch.cfloat diff --git a/test/torchaudio_unittest/torchscript_consistency_impl.py b/test/torchaudio_unittest/torchscript_consistency_impl.py index e7d5849b72..5b84ab897c 100644 --- a/test/torchaudio_unittest/torchscript_consistency_impl.py +++ b/test/torchaudio_unittest/torchscript_consistency_impl.py @@ -550,6 +550,7 @@ def test_TimeStretch(self): tensor, ) + class Transforms(common_utils.TestBaseMixin): """Implements test for Transforms that are performed for different devices""" def _assert_consistency(self, transform, tensor):