fix

zhengruifeng · Jan 25, 2025 · 8475f41 · 8475f41
1 parent 23b3b82
commit 8475f41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pyspark/ml/tests/test_feature.py b/python/pyspark/ml/tests/test_feature.py
@@ -574,7 +574,7 @@ def test_imputer(self):
         self.assertEqual(output.count(), 5)
 
         # save & load
-        with tempfile.TemporaryDirectory(prefix="word2vec") as d:
+        with tempfile.TemporaryDirectory(prefix="imputer") as d:
             imputer.write().overwrite().save(d)
             imputer2 = Imputer.load(d)
             self.assertEqual(str(imputer), str(imputer2))