diff --git a/gateway_provisioners/yarn.py b/gateway_provisioners/yarn.py index 40e194f..45e97d0 100644 --- a/gateway_provisioners/yarn.py +++ b/gateway_provisioners/yarn.py @@ -587,15 +587,16 @@ def _query_app_by_name(self, kernel_id: str) -> dict | None: ) else: data = response.data - if ( - isinstance(type, dict) - and isinstance(data.get("apps"), dict) - and "app" in data.get("apps") - ): - for app in data["apps"]["app"]: - if app.get("name", "").find(kernel_id) >= 0 and app.get("id") > top_most_app_id: - target_app = app - top_most_app_id = app.get("id") + if isinstance(data, dict): + apps = data.get("apps") + if isinstance(apps, dict) and "app" in apps: + for app in data["apps"]["app"]: + if ( + app.get("name", "").find(kernel_id) >= 0 + and app.get("id") > top_most_app_id + ): + target_app = app + top_most_app_id = app.get("id") return target_app def _query_app_by_id(self, app_id: str) -> dict | None: diff --git a/tests/test_yarn_provisioners.py b/tests/test_yarn_provisioners.py index 9fdb9e3..984f719 100644 --- a/tests/test_yarn_provisioners.py +++ b/tests/test_yarn_provisioners.py @@ -16,7 +16,6 @@ @pytest.mark.parametrize("seed_env", [YARN_SEED_ENV]) -@pytest.mark.xfail() async def test_lifecycle(init_api_mocks, response_manager, get_provisioner, seed_env): name = "yarn" kernel_id = str(uuid4())