Skip to content

Commit

Permalink
add import
Browse files Browse the repository at this point in the history
  • Loading branch information
anjali411 committed Jul 2, 2020
1 parent 75fb528 commit 1ebd7e7
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 1 deletion.
3 changes: 2 additions & 1 deletion test/torchscript_consistency_cuda_test.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch

from . import common_utils
from .torchscript_consistency_impl import Functional, Transforms
from .torchscript_consistency_impl import Functional, Transforms, TransformsWithComplexDtypes


@common_utils.skipIfNoCuda
Expand All @@ -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
Expand Down
1 change: 1 addition & 0 deletions test/torchscript_consistency_impl.py
Original file line number Diff line number Diff line change
Expand Up @@ -551,6 +551,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

0 comments on commit 1ebd7e7

Please sign in to comment.