CU-86b2czk01 Fixing linter issues

sdgym/synthesizers/realtabformer.py

@@ -1,10 +1,13 @@
 """REaLTabFormer integration."""
+
 import contextlib
+from functools import partialmethod
+
 import tqdm
 
-from functools import partialmethod
 from sdgym.synthesizers.base import BaselineSynthesizer
 
+
 @contextlib.contextmanager
 def prevent_tqdm_output():
     """Temporarily disables tqdm m."""

tests/unit/synthesizers/test_realtabformer.py

@@ -25,7 +25,7 @@ class TestRealTabFormerSynthesizer:
 
     def test_get_trained_synthesizer(self):
         """Test _get_trained_synthesizer initializes
-           and fits REaLTabFormer with correct parameters."""
+        and fits REaLTabFormer with correct parameters."""
         with patch('realtabformer.REaLTabFormer') as MockREaLTabFormer:
             # Setup
             mock_model = MagicMock()
@@ -40,7 +40,7 @@ def test_get_trained_synthesizer(self):
             # Assert
             MockREaLTabFormer.assert_called_once_with(model_type='tabular')
             mock_model.fit.assert_called_once_with(data, device='cpu')
-            assert result == mock_model, "Expected the trained model to be returned."
+            assert result == mock_model, 'Expected the trained model to be returned.'
 
     def test_sample_from_synthesizer(self):
         """Test _sample_from_synthesizer generates data with the specified sample size."""
@@ -55,5 +55,6 @@ def test_sample_from_synthesizer(self):
 
         # Assert
         trained_model.sample.assert_called_once_with(n_sample, device='cpu')
-        assert synthetic_data.shape[0] == n_sample, \
-            f"Expected {n_sample} rows, but got {synthetic_data.shape[0]}"
+        assert synthetic_data.shape[0] == n_sample, (
+            f'Expected {n_sample} rows, but got {synthetic_data.shape[0]}'
+        )