From 89ba39b726a0d28e5074592a255e45ef9749204a Mon Sep 17 00:00:00 2001 From: anjali411 Date: Wed, 1 Jul 2020 16:49:56 -0700 Subject: [PATCH] add import --- test/torchscript_consistency_cuda_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/torchscript_consistency_cuda_test.py b/test/torchscript_consistency_cuda_test.py index 49ac5a64ddf..afc09010ee0 100644 --- a/test/torchscript_consistency_cuda_test.py +++ b/test/torchscript_consistency_cuda_test.py @@ -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