format

Signed-off-by: Achal Shah <[email protected]>
feast-dev · Jun 9, 2021 · 44b07eb · 44b07eb
1 parent 41136c8
commit 44b07eb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/sdk/python/feast/errors.py b/sdk/python/feast/errors.py
@@ -80,7 +80,9 @@ def __init__(self, expected, missing):
 
 
 class FeastJoinKeysDuringMaterialization(Exception):
-    def __init__(self, source: str, join_key_columns: Set[str], source_columns: Set[str]):
+    def __init__(
+        self, source: str, join_key_columns: Set[str], source_columns: Set[str]
+    ):
         super().__init__(
             f"The DataFrame from {source} being materialized must have at least {join_key_columns} columns present, "
             f"but these were missing: {join_key_columns - source_columns} "

diff --git a/sdk/python/feast/infra/offline_stores/file.py b/sdk/python/feast/infra/offline_stores/file.py
@@ -220,8 +220,7 @@ def pull_latest_from_table_or_query(
         source_columns = set(source_df.columns)
         if not set(join_key_columns).issubset(source_columns):
             raise FeastJoinKeysDuringMaterialization(
-                data_source.path,
-                set(join_key_columns), source_columns
+                data_source.path, set(join_key_columns), source_columns
             )
 
         ts_columns = (