Skip to content
Merged
19 changes: 10 additions & 9 deletions gateway_provisioners/yarn.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
1 change: 0 additions & 1 deletion tests/test_yarn_provisioners.py
Original file line number Diff line number Diff line change
Expand Up @@ -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())
Expand Down