Addressed comments

zou3519 · zou3519 · commit fabd77594df5 · 2018-05-07T07:40:55.000-07:00
diff --git a/test/test_torch.py b/test/test_torch.py
@@ -3374,7 +3374,8 @@ def _test_gesv_batched(self, cast):
         # Test non-contiguous inputs.
         if not TEST_NUMPY:
             return
-
+        import numpy
+        from numpy.linalg import solve
         A = cast(torch.randn(2, 2, 2)).permute(1, 0, 2)
         b = cast(torch.randn(2, 2, 2)).permute(2, 1, 0)
         x, _ = torch.gesv(b, A)
