add import

pytorch · Aug 6, 2020 · 4509fe8 · 4509fe8
1 parent 7f98428
commit 4509fe8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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):
-Original file line number
+Diff line change
@@ Expand Up / @@ -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):
@@ Expand Down @@