Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix
Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
  • Loading branch information
kevjumba committed Jul 29, 2022
commit 7b6ee9ff58380318ee8a3548d33777b019a9cfea
4 changes: 2 additions & 2 deletions sdk/python/tests/utils/feature_test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
from feast.protos.feast.serving.ServingService_pb2 import FieldStatus


def get_latest_rows(df, join_key, entity_values):
def get_latest_rows(df: pd.DataFrame, join_key: str, entity_values: List[str]) -> pd.DataFrame:
"""
Return latest rows in a dataframe based on join key and entity values.

Expand All @@ -38,7 +38,7 @@ def generate_expected_logs(
features: List[str],
join_keys: List[str],
timestamp_column: str,
):
) -> pd.DataFrame:
"""
Given dataframe and feature view, generate the expected logging dataframes that would be otherwise generated by our logging infrastructure.
Args:
Expand Down