diff --git a/tests/unit/multi_table/test_base.py b/tests/unit/multi_table/test_base.py index 21930ad04..3e3abdbb9 100644 --- a/tests/unit/multi_table/test_base.py +++ b/tests/unit/multi_table/test_base.py @@ -1,4 +1,5 @@ import logging +import os import re import warnings from collections import defaultdict @@ -1533,7 +1534,7 @@ def test_save_warning(self, tmp_path): 'first and then saving.' ) with pytest.warns(Warning, match=warn_msg): - filepath = tmp_path / 'output.pkl' + filepath = os.path.join(tmp_path, 'output.pkl') synthesizer.save(filepath) @patch('sdv.multi_table.base.datetime') diff --git a/tests/unit/single_table/test_base.py b/tests/unit/single_table/test_base.py index 497921c36..3830f3f4f 100644 --- a/tests/unit/single_table/test_base.py +++ b/tests/unit/single_table/test_base.py @@ -1,4 +1,5 @@ import logging +import os import re from datetime import date, datetime from unittest.mock import ANY, MagicMock, Mock, call, mock_open, patch @@ -1821,7 +1822,7 @@ def test_save_warning(self, tmp_path): 'first and then saving.' ) with pytest.warns(Warning, match=warn_msg): - filepath = tmp_path / 'output.pkl' + filepath = os.path.join(tmp_path, 'output.pkl') synthesizer.save(filepath) @patch('sdv.single_table.base.datetime')