add import

pytorch · Jul 1, 2020 · 89ba39b · 89ba39b
1 parent 75fb528
commit 89ba39b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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