From 41a3e6a9ca0bd04d46f397f2d69f15c82f15c311 Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <jgbt@protonmail.com>
Date: Sun, 5 Dec 2021 20:00:40 -0500
Subject: [PATCH] Type check fixes

---
 tractor/_actor.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tractor/_actor.py b/tractor/_actor.py
index b78b872..1f7d9f5 100644
--- a/tractor/_actor.py
+++ b/tractor/_actor.py
@@ -182,6 +182,7 @@ async def _invoke(
                 else:
                     raise
 
+            assert cs
             if cs.cancelled_caught or ctx._error:
 
                 # TODO: pack in ``trio.Cancelled.__traceback__`` here
@@ -706,7 +707,7 @@ class Actor:
             ctx = self._contexts[(actor_uid, cid)]
 
             # adjust buffer size if specified
-            state = ctx._send_chan._state
+            state = ctx._send_chan._state  # type: ignore
             if msg_buffer_size and state.max_buffer_size != msg_buffer_size:
                 state.max_buffer_size = msg_buffer_size