Fix lint issues

Author: jcatt-sf

src/datacustomcode/einstein_predictions/spark_base.py

@@ -22,12 +22,13 @@
     Optional,
 )
 
-from datacustomcode.einstein_predictions.types import PredictionType
 from datacustomcode.mixin import UserExtendableNamedConfigMixin
 
 if TYPE_CHECKING:
     from pyspark.sql import Column
 
+    from datacustomcode.einstein_predictions.types import PredictionType
+
 
 class SparkEinsteinPredictions(ABC, UserExtendableNamedConfigMixin):
     CONFIG_NAME: str

src/datacustomcode/einstein_predictions/spark_default.py

@@ -159,8 +159,7 @@ def _build_request(
     settings: Optional[Dict[str, Any]],
 ) -> PredictionRequest:
     prediction_columns = [
-        _feature_to_prediction_column(name, value)
-        for name, value in features.items()
+        _feature_to_prediction_column(name, value) for name, value in features.items()
     ]
     return PredictionRequest(
         prediction_type=prediction_type,

tests/test_spark_einstein_predictions.py

@@ -17,10 +17,7 @@
     _invoke_predictions,
     _invoke_predictions_as_struct,
 )
-from datacustomcode.einstein_predictions.types import (
-    PredictionResponse,
-    PredictionType,
-)
+from datacustomcode.einstein_predictions.types import PredictionResponse, PredictionType
 
 
 def _success_response(data: dict | None = None) -> PredictionResponse:
@@ -76,7 +73,7 @@ def test_returns_object_from_config(self):
             )
 
             assert _build_underlying_predictions() is mock_predictions
-            mock_obj_config.einstein_predictions_config.to_object.assert_called_once_with()  # noqa: E501
+            mock_obj_config.einstein_predictions_config.to_object.assert_called_once_with()
 
     def test_raises_when_config_missing(self):
         with patch(