Skip to content

Commit

Permalink
Feedback
Browse files Browse the repository at this point in the history
  • Loading branch information
fealho committed Jul 26, 2024
1 parent fcd821b commit d0e0789
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
3 changes: 2 additions & 1 deletion tests/unit/multi_table/test_base.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import logging
import os
import re
import warnings
from collections import defaultdict
Expand Down Expand Up @@ -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')
Expand Down
3 changes: 2 additions & 1 deletion tests/unit/single_table/test_base.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -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')
Expand Down

0 comments on commit d0e0789

Please sign in to comment.