From 44b07eb7cca781629c7e79d68292c783a8c58276 Mon Sep 17 00:00:00 2001 From: Achal Shah Date: Wed, 9 Jun 2021 11:54:10 -0700 Subject: [PATCH] format Signed-off-by: Achal Shah --- sdk/python/feast/errors.py | 4 +++- sdk/python/feast/infra/offline_stores/file.py | 3 +-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/sdk/python/feast/errors.py b/sdk/python/feast/errors.py index 5eeec6a07f..b186d9ab33 100644 --- 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 index 171512c851..acd12ff900 100644 --- 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 = (