minor

apache · Sep 22, 2018 · 01bb209 · 01bb209
1 parent 4d114fc
commit 01bb209
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/pyspark/context.py b/python/pyspark/context.py
@@ -537,8 +537,10 @@ def _serialize_to_jvm(self, data, serializer, reader_func, createRDDServer):
             # parallelize from there.
             tempFile = NamedTemporaryFile(delete=False, dir=self._temp_dir)
             try:
-                serializer.dump_stream(data, tempFile)
-                tempFile.close()
+                try:
+                    serializer.dump_stream(data, tempFile)
+                finally:
+                    tempFile.close()
                 return reader_func(tempFile.name)
             finally:
                 # we eagerily reads the file so we can delete right after.