txl_kernel 0.1.17__tar.gz → 0.1.20__tar.gz

{txl_kernel-0.1.17 → txl_kernel-0.1.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: txl_kernel
-Version: 0.1.17
+Version: 0.1.20
 Summary: TXL plugin for a kernel driver
 Project-URL: Source, https://github.com/davidbrochart/jpterm/plugins/kernel
 Author-email: David Brochart <david.brochart@gmail.com>

txl_kernel-0.1.20/txl_kernel/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.20"

{txl_kernel-0.1.17 → txl_kernel-0.1.20}/txl_kernel/driver.py

@@ -28,9 +28,7 @@ class Comm:
             buffers=buffers,
         )
         self.msg_cnt += 1
-        asyncio.create_task(
-            self.send_message(msg, self.shell_channel, change_date_to_str=True)
-        )
+        asyncio.create_task(self.send_message(msg, self.shell_channel, change_date_to_str=True))
 
 
 class KernelMixin:
@@ -61,18 +59,14 @@ class KernelMixin:
                 "shell": [asyncio.Future()],
             }
             try:
-                msg = await asyncio.wait_for(
-                    self.execute_requests[msg_id]["shell"][0], new_timeout
-                )
+                msg = await asyncio.wait_for(self.execute_requests[msg_id]["shell"][0], new_timeout)
             except asyncio.TimeoutError:
                 del self.execute_requests[msg_id]
                 error_message = f"Kernel didn't respond in {timeout} seconds"
                 raise RuntimeError(error_message)
             if msg["header"]["msg_type"] == "kernel_info_reply":
                 try:
-                    msg = await asyncio.wait_for(
-                        self.execute_requests[msg_id]["iopub"][0], 0.2
-                    )
+                    msg = await asyncio.wait_for(self.execute_requests[msg_id]["iopub"][0], 0.2)
                 except asyncio.TimeoutError:
                     pass
                 else:
@@ -88,9 +82,7 @@ class KernelMixin:
             if msg_type == "comm_open":
                 for comm_handler in self.comm_handlers:
                     comm_id = msg["content"]["comm_id"]
-                    comm = Comm(
-                        comm_id, self.shell_channel, self.session_id, self.send_message
-                    )
+                    comm = Comm(comm_id, self.shell_channel, self.session_id, self.send_message)
                     comm_handler.comm_open(msg, comm)
             elif msg_type == "comm_msg":
                 for comm_handler in self.comm_handlers:
@@ -150,9 +142,7 @@ class KernelMixin:
                 del self.execute_requests[msg_id]
                 error_message = f"Kernel didn't respond in {timeout} seconds"
                 raise RuntimeError(error_message)
-            await self._handle_outputs(
-                ydoc, ycell, self.execute_requests[msg_id]["iopub"]
-            )
+            await self._handle_outputs(ydoc, ycell, self.execute_requests[msg_id]["iopub"])
             try:
                 await asyncio.wait_for(
                     self.execute_requests[msg_id]["shell"][0],
@@ -186,9 +176,7 @@ class KernelMixin:
             outputs = list(ycell["outputs"])
             if msg_type == "stream":
                 if (not outputs) or (outputs[-1]["name"] != content["name"]):
-                    outputs.append(
-                        {"name": content["name"], "output_type": msg_type, "text": []}
-                    )
+                    outputs.append({"name": content["name"], "output_type": msg_type, "text": []})
                 outputs[-1]["text"].append(content["text"])
             elif msg_type in ("display_data", "execute_result"):
                 outputs.append(

{txl_kernel-0.1.17 → txl_kernel-0.1.20}/txl_kernel/message.py

@@ -12,7 +12,7 @@ def utcnow() -> datetime:
 
 
 def date_to_str(obj: Dict[str, Any]):
-    if "date" in obj and type(obj["date"]) is not str:
+    if "date" in obj and not isinstance(obj["date"], str):
         obj["date"] = obj["date"].isoformat().replace("+00:00", "Z")
     return obj
 
@@ -23,9 +23,7 @@ def str_to_date(obj: Dict[str, Any]) -> Dict[str, Any]:
     return obj
 
 
-def create_message_header(
-    msg_type: str, session_id: str, msg_id: str
-) -> Dict[str, Any]:
+def create_message_header(msg_type: str, session_id: str, msg_id: str) -> Dict[str, Any]:
     if not session_id:
         session_id = msg_id = uuid4().hex
     else:

txl_kernel-0.1.17/txl_kernel/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.1.17"