Skip to content

Commit

Permalink
Avoid key error if frontend is not supported as a backend.
Browse files Browse the repository at this point in the history
  • Loading branch information
ReneFabricius committed Sep 14, 2023
1 parent 44ac5bb commit 1ad40c6
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -226,7 +226,9 @@ def get_dtypes(

# If being called from a frontend test
if test_globals.CURRENT_FRONTEND is not test_globals._Notsetval:
frontend_dtypes = _get_type_dict(test_globals.CURRENT_FRONTEND, kind, True)
frontend_dtypes = _get_type_dict_helper(
test_globals.CURRENT_FRONTEND, kind, True
)
valid_dtypes = valid_dtypes.intersection(frontend_dtypes)

# Make sure we return dtypes that are compatible with ground truth backend
Expand Down

0 comments on commit 1ad40c6

Please sign in to comment.