Fix on bug causing Kernel launch to timeout (#141)

Author: rao23

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:

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())