Fix a test.

apache · Nov 21, 2017 · d741171 · d741171
1 parent 3e23653
commit d741171
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/pyspark/sql/tests.py b/python/pyspark/sql/tests.py
@@ -2587,6 +2587,7 @@ def _to_pandas(self):
 
     @unittest.skipIf(not _have_pandas, "Pandas not installed")
     def test_to_pandas(self):
+        import numpy as np
         pdf = self._to_pandas()
         types = pdf.dtypes
         self.assertEquals(types[0], np.int32)