Fix exception typing

bi_streaming
Tyler Goodlet 2021-06-13 20:19:52 -04:00
parent 0af58522a4
commit 008314554c
1 changed files with 3 additions and 3 deletions

View File

@ -1,7 +1,7 @@
"""
Our classy exception set.
"""
from typing import Dict, Any, Optional
from typing import Dict, Any, Optional, Type
import importlib
import builtins
import traceback
@ -18,7 +18,7 @@ class RemoteActorError(Exception):
def __init__(
self,
message: str,
suberror_type: Optional[Exception] = None,
suberror_type: Optional[Type[BaseException]] = None,
**msgdata
) -> None:
@ -84,7 +84,7 @@ def unpack_error(
tb_str = error.get('tb_str', '')
message = f"{chan.uid}\n" + tb_str
type_name = error['type_str']
suberror_type = Exception
suberror_type: Type[BaseException] = Exception
if type_name == 'ContextCancelled':
err_type = ContextCancelled