fix lint

sdv-dev · Sep 12, 2024 · 7f66113 · 7f66113
1 parent 6ed0260
commit 7f66113
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/sdv/metadata/metadata.py b/sdv/metadata/metadata.py
@@ -35,8 +35,8 @@ def load_from_json(cls, filepath, single_table_name=None):
         if metadata.get('METADATA_SPEC_VERSION') == 'SINGLE_TABLE_V1':
             single_table_name = single_table_name or cls.DEFAULT_SINGLE_TABLE_NAME
             warnings.warn(
-                'You are loading an older SingleTableMetadata object. This will be converted '
-                f"into the new Metadata object with a placeholder table name ('{single_table_name}')."
+                'You are loading an older SingleTableMetadata object. This will be converted into'
+                f" the new Metadata object with a placeholder table name ('{single_table_name}')."
                 ' Please save this new object for future usage.'
             )
 

diff --git a/tests/unit/metadata/test_metadata.py b/tests/unit/metadata/test_metadata.py
@@ -1,5 +1,5 @@
 import re
-from unittest.mock import call, Mock, patch
+from unittest.mock import Mock, call, patch
 
 import pandas as pd
 import pytest
@@ -122,7 +122,9 @@ def test_load_from_json_single_table(self, mock_read_json):
         )
 
         expected_warning_with_table_name = re.escape(warning_message.format('filepath'))
-        expected_warning_without_table_name = re.escape(warning_message.format('default_table_name'))
+        expected_warning_without_table_name = re.escape(
+            warning_message.format('default_table_name')
+        )
 
         # Run
         with pytest.warns(UserWarning, match=expected_warning_with_table_name):