fmt

Author: nerdai

tests/clients/test_flexible_client.py

@@ -24,6 +24,7 @@
 from tests.test_utils.assert_metrics_dict import assert_metrics_dict
 from tests.test_utils.models_for_test import LinearModel
 
+
 freezegun.configure(extend_ignore_list=["transformers"])  # type: ignore
 
 
@@ -249,8 +250,7 @@ def mock_validate_or_test(  # type: ignore
             fold_loss_dict_into_metrics(self.mock_metrics, self.mock_loss_dict, logging_mode)
         if logging_mode == LoggingMode.VALIDATION:
             return self.mock_loss, self.mock_metrics
-        else:
-            return self.mock_loss, self.mock_metrics_test
+        return self.mock_loss, self.mock_metrics_test
 
 
 def test_subclass_raises_warning_if_override_val_step() -> None:
@@ -261,7 +261,7 @@ def test_subclass_raises_warning_if_override_val_step() -> None:
     with pytest.warns(RuntimeWarning, match=re.escape(msg)):
 
         class _TestSubclass(FlexibleClient):
-            """A subclass that overrides val_step"""
+            """A subclass that overrides val_step."""
 
             def val_step(
                 self, input: TorchInputType, target: TorchTargetType
@@ -278,7 +278,7 @@ def test_subclass_raises_warning_if_override_train_step() -> None:
     with pytest.warns(RuntimeWarning, match=re.escape(msg)):
 
         class _TestSubclass(FlexibleClient):
-            """A subclass that overrides train_step"""
+            """A subclass that overrides train_step."""
 
             def train_step(
                 self, input: TorchInputType, target: TorchTargetType

tests/mixins/personalized/test_ditto.py

@@ -49,7 +49,6 @@ class _TestDittoedClient(DittoPersonalizedMixin, _TestFlexibleClient):
 
 
 class _DummyParent:
-
     def __init__(self) -> None:
         pass
 
@@ -81,15 +80,14 @@ class _InvalidTestDittoClient(DittoPersonalizedMixin):
         pass
 
     with pytest.raises(RuntimeError, match="This object needs to satisfy `FlexibleClientProtocolPreSetup`."):
-
         _InvalidTestDittoClient(data_path=Path(""), metrics=[Accuracy()])
 
 
 def test_subclass_checks_raise_no_warning() -> None:
     with warnings.catch_warnings(record=True) as recorded_warnings:
 
         class _TestInheritanceMixin(DittoPersonalizedMixin, _TestFlexibleClient):
-            """subclass should skip validation if is itself a Mixin that inherits DittoPersonalizedMixin"""
+            """Subclass should skip validation if is itself a Mixin that inherits DittoPersonalizedMixin."""
 
             pass
 
@@ -370,7 +368,6 @@ def test_val_step(
 
 def test_raise_runtime_error_not_flexible_client() -> None:
     """Test that an invalid parent raises RuntimeError."""
-
     with pytest.raises(
         RuntimeError, match=re.escape("This object needs to satisfy `FlexibleClientProtocolPreSetup`.")
     ):

tests/mixins/test_adaptive_drift_constrained.py

@@ -67,15 +67,14 @@ class _InvalidTestAdaptedClient(AdaptiveDriftConstrainedMixin):
         pass
 
     with pytest.raises(RuntimeError, match="This object needs to satisfy `FlexibleClientProtocolPreSetup`."):
-
         _InvalidTestAdaptedClient(data_path=Path(""), metrics=[Accuracy()])
 
 
 def test_subclass_checks_raise_no_warning() -> None:
     with warnings.catch_warnings(record=True) as recorded_warnings:
 
         class _TestInheritanceMixin(AdaptiveDriftConstrainedMixin, _TestFlexibleClient):
-            """subclass should skip validation if is itself a Mixin that inherits AdaptiveDriftConstrainedMixin"""
+            """Subclass should skip validation if is itself a Mixin that inherits AdaptiveDriftConstrainedMixin."""
 
             pass