Use rmtree, instead of removedirs

mengxr · Jul 2, 2015 · 0ef58b6 · 0ef58b6
1 parent cb21653
commit 0ef58b6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/pyspark/mllib/feature.py b/python/pyspark/mllib/feature.py
@@ -502,8 +502,9 @@ class Word2Vec(object):
     >>> sameModel = Word2VecModel.load(sc, path)
     >>> model.transform("a") == sameModel.transform("a")
     True
+    >>> from shutil import rmtree
     >>> try:
-    ...     os.removedirs(path)
+    ...     rmtree(path)
     ... except OSError:
     ...     pass
     """