Skip to content

Commit

Permalink
Fix integration tests
Browse files Browse the repository at this point in the history
Signed-off-by: Tsotne Tabidze <tsotne@tecton.ai>
  • Loading branch information
Tsotne Tabidze committed Jan 14, 2022
1 parent 6a67d76 commit eedac3c
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -212,11 +212,11 @@ def _get_online_features_dict_remotely(
time.sleep(1)
else:
raise Exception("Failed to get online features from remote feature server")
keys = response["field_values"][0]["statuses"].keys()
keys = response["metadata"]["feature_names"]
# Get rid of unnecessary structure in the response, leaving list of dicts
response = [row["fields"] for row in response["field_values"]]
response = [row["values"] for row in response["results"]]
# Convert list of dicts (response) into dict of lists which is the format of the return value
return {key: [row.get(key) for row in response] for key in keys}
return {key: [row[idx] for row in response] for idx, key in enumerate(keys)}


def get_online_features_dict(
Expand Down

0 comments on commit eedac3c

Please sign in to comment.