diff --git a/examples/team_recommender/tests/example_6_n_generations/test_faster_with_n_generations.py b/examples/team_recommender/tests/example_6_n_generations/test_faster_with_n_generations.py index 9252e92..0454957 100644 --- a/examples/team_recommender/tests/example_6_n_generations/test_faster_with_n_generations.py +++ b/examples/team_recommender/tests/example_6_n_generations/test_faster_with_n_generations.py @@ -17,7 +17,10 @@ def get_all_developer_names(skills_data) -> set[str]: def get_developer_names_from_response(response) -> set[str]: - return {developer["name"] for developer in response["developers"]} + if "developers" in response: + return {developer["name"] for developer in response["developers"]} + else: + return set() def has_expected_success_rate(results: list[bool], expected_success_rate: float) -> bool: