check env in kwargs

jupyter · Jun 21, 2023 · 623b2b3 · 623b2b3
1 parent 261af36
commit 623b2b3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/jupyter_client/provisioning/local_provisioner.py b/jupyter_client/provisioning/local_provisioner.py
@@ -186,8 +186,11 @@ async def pre_launch(self, **kwargs: Any) -> Dict[str, Any]:
                 km.hb_port = lpc.find_available_port(km.ip)
                 km.control_port = lpc.find_available_port(km.ip)
                 self.ports_cached = True
-
-            km.write_connection_file(jupyter_session=kwargs['env'].get("JPY_SESSION_NAME", ""))
+            if 'env' in kwargs:
+                jupyter_session = kwargs['env'].get("JPY_SESSION_NAME", "")
+                km.write_connection_file(jupyter_session=jupyter_session)
+            else:
+                km.write_connection_file()
             self.connection_info = km.get_connection_info()
 
             kernel_cmd = km.format_kernel_cmd(