From 1ad40c6e33e7352a77212499afe64b2e4051a13d Mon Sep 17 00:00:00 2001 From: ReneFabricius <65137969+ReneFabricius@users.noreply.github.com> Date: Thu, 14 Sep 2023 12:40:55 +0300 Subject: [PATCH] Avoid key error if frontend is not supported as a backend. --- .../test_ivy/helpers/hypothesis_helpers/dtype_helpers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ivy_tests/test_ivy/helpers/hypothesis_helpers/dtype_helpers.py b/ivy_tests/test_ivy/helpers/hypothesis_helpers/dtype_helpers.py index e2c8fa6cdf5b8..c7ba3496a2c9a 100644 --- a/ivy_tests/test_ivy/helpers/hypothesis_helpers/dtype_helpers.py +++ b/ivy_tests/test_ivy/helpers/hypothesis_helpers/dtype_helpers.py @@ -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