lint fixes

Signed-off-by: Calvin Hopkins <[email protected]>

Author: thechopkins

sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/spark_source.py

@@ -191,11 +191,10 @@ def __hash__(self):
 
     def __eq__(self, other):
         if not isinstance(other, SparkSource):
-            raise TypeError("Comparisons should only involve SparkSource class objects.")
-        return (
-            super().__eq__(other)
-            and self.spark_options == other.spark_options
-        )
+            raise TypeError(
+                "Comparisons should only involve SparkSource class objects."
+            )
+        return super().__eq__(other) and self.spark_options == other.spark_options
 
 
 class SparkOptions:
@@ -296,7 +295,9 @@ def to_proto(self) -> DataSourceProto.SparkOptions:
 
     def __eq__(self, other: object) -> bool:
         if not isinstance(other, SparkOptions):
-            raise TypeError("Comparisons should only involve SparkOptions class objects.")
+            raise TypeError(
+                "Comparisons should only involve SparkOptions class objects."
+            )
 
         return (
             self.table == other.table

sdk/python/tests/unit/test_data_sources.py

@@ -10,10 +10,12 @@
 )
 from feast.field import Field
 from feast.infra.offline_stores.bigquery_source import BigQuerySource
+from feast.infra.offline_stores.contrib.spark_offline_store.spark_source import (
+    SparkSource,
+)
 from feast.infra.offline_stores.file_source import FileSource
 from feast.infra.offline_stores.redshift_source import RedshiftSource
 from feast.infra.offline_stores.snowflake_source import SnowflakeSource
-from feast.infra.offline_stores.contrib.spark_offline_store.spark_source import SparkSource
 from feast.types import Bool, Float32, Int64
 
 
@@ -235,52 +237,34 @@ def test_redshift_fully_qualified_table_name(source_kwargs, expected_name):
 
     assert redshift_source.redshift_options.fully_qualified_table_name == expected_name
 
+
 @pytest.mark.parameterize(
-        "test_data,are_equal",
-        [
-            (
-                SparkSource(
-                    name='name',
-                    table='table',
-                    query='query',
-                    file_format='file_format'
-                ),
-                True
-            ),
-            (
-                SparkSource(
-                    table='table',
-                    query='query',
-                    file_format='file_format'
-                ),
-                False
+    "test_data,are_equal",
+    [
+        (
+            SparkSource(
+                name="name", table="table", query="query", file_format="file_format"
             ),
-            (
-                SparkSource(
-                    name='name',
-                    table='table',
-                    query='query',
-                    file_format='file_format1'
-                ),
-                False
+            True,
+        ),
+        (SparkSource(table="table", query="query", file_format="file_format"), False),
+        (
+            SparkSource(
+                name="name", table="table", query="query", file_format="file_format1"
             ),
-            (
-                SparkSource(
-                    name='name',
-                    table='table',
-                    query='query1',
-                    file_format='file_format'
-                ),
-                True
+            False,
+        ),
+        (
+            SparkSource(
+                name="name", table="table", query="query1", file_format="file_format"
             ),
-        ]
+            True,
+        ),
+    ],
 )
 def test_spark_source_equality(test_data, are_equal):
     default = SparkSource(
-        name='name',
-        table='table1',
-        query='query',
-        file_format='file_format'
+        name="name", table="table1", query="query", file_format="file_format"
     )
     if are_equal:
         assert default == test_data