Remove unnecessary pickle file generated while testing (#732)

sdv-dev · Oct 25, 2023 · 2ee0077 · 2ee0077
1 parent 8d6ec14
commit 2ee0077
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration/transformers/test_categorical.py b/tests/integration/transformers/test_categorical.py
@@ -452,9 +452,9 @@ def test_one_hot_doesnt_warn(tmp_path):
     # Run
     ohe.fit(data, 'column_name')
     tmp = tmp_path / 'ohe.pkl'
-    with open(tmp.name, 'wb') as f:
+    with open(tmp, 'wb') as f:
         pickle.dump(ohe, f)
-    with open(tmp.name, 'rb') as f:
+    with open(tmp, 'rb') as f:
         ohe_loaded = pickle.load(f)
 
     # Assert