dara-core 1.21.16__py3-none-any.whl → 1.21.17__py3-none-any.whl

dara/core/internal/dependency_resolution.py

@@ -15,9 +15,9 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from typing import Any, List, Literal, Optional, Union
+from typing import Any, Literal, TypeGuard
 
-from typing_extensions import TypedDict, TypeGuard
+from typing_extensions import TypedDict
 
 from dara.core.base_definitions import BaseTask, PendingTask
 from dara.core.interactivity import DerivedVariable
@@ -31,8 +31,8 @@ from dara.core.logging import dev_logger
 class ResolvedDerivedVariable(TypedDict):
     type: Literal['derived']
     uid: str
-    values: List[Any]
-    force_key: Optional[str]
+    values: list[Any]
+    force_key: str | None
 
 
 class ResolvedServerVariable(TypedDict):
@@ -90,11 +90,7 @@ def clean_force_key(value: Any) -> Any:
 
 
 async def resolve_dependency(
-    entry: Union[
-        ResolvedDerivedVariable,
-        ResolvedSwitchVariable,
-        Any,
-    ],
+    entry: ResolvedDerivedVariable | ResolvedSwitchVariable | Any,
     store: CacheStore,
     task_mgr: TaskManager,
 ):
@@ -135,7 +131,7 @@ async def _resolve_derived_var(
 
     registry_mgr: RegistryLookup = utils_registry.get('RegistryLookup')
     var = await registry_mgr.get(derived_variable_registry, str(derived_variable_entry.get('uid')))
-    input_values: List[Any] = derived_variable_entry.get('values', [])
+    input_values: list[Any] = derived_variable_entry.get('values', [])
     result = await DerivedVariable.get_value(
         var_entry=var,
         store=store,

dara/core/internal/devtools.py

@@ -21,13 +21,12 @@ import sys
 import traceback
 from contextlib import contextmanager
 from datetime import datetime
-from typing import Optional
 
 from dara.core.internal.websocket import WebsocketManager
 from dara.core.logging import eng_logger
 
 
-def print_stacktrace(err: Optional[BaseException] = None) -> str:
+def print_stacktrace(err: BaseException | None = None) -> str:
     """
     Prints out the current stack trace. Will also extract any exceptions and print them at the end.
     """
@@ -59,7 +58,7 @@ def handle_system_exit(error_msg: str):
         raise InterruptedError(error_msg) from e
 
 
-def get_error_for_channel(err: Optional[BaseException] = None) -> dict:
+def get_error_for_channel(err: BaseException | None = None) -> dict:
     """
     Get error from current stacktrace to send to the client
     """

dara/core/internal/download.py

@@ -18,9 +18,8 @@ limitations under the License.
 from __future__ import annotations
 
 import os
-from collections.abc import Awaitable
+from collections.abc import Awaitable, Callable
 from contextvars import ContextVar
-from typing import Callable, Optional, Tuple
 from uuid import uuid4
 
 import anyio
@@ -38,8 +37,8 @@ class DownloadDataEntry(BaseModel):
     uid: str
     file_path: str
     cleanup_file: bool
-    identity_name: Optional[str] = None
-    download: Callable[[DownloadDataEntry], Awaitable[Tuple[anyio.AsyncFile, Callable[..., Awaitable]]]]
+    identity_name: str | None = None
+    download: Callable[[DownloadDataEntry], Awaitable[tuple[anyio.AsyncFile, Callable[..., Awaitable]]]]
     """Handler for getting the file from the entry"""
 
 
@@ -48,7 +47,7 @@ DownloadRegistryEntry = CachedRegistryEntry(
 )  # expire the codes after 10 minutes
 
 
-async def download(data_entry: DownloadDataEntry) -> Tuple[anyio.AsyncFile, Callable[..., Awaitable]]:
+async def download(data_entry: DownloadDataEntry) -> tuple[anyio.AsyncFile, Callable[..., Awaitable]]:
     """
     Get the loaded filename and path from a code
 
@@ -74,7 +73,7 @@ async def download(data_entry: DownloadDataEntry) -> Tuple[anyio.AsyncFile, Call
     return (async_file, cleanup)
 
 
-GENERATE_CODE_OVERRIDE = ContextVar[Optional[Callable[[str], str]]]('GENERATE_CODE_OVERRIDE', default=None)
+GENERATE_CODE_OVERRIDE = ContextVar[Callable[[str], str] | None]('GENERATE_CODE_OVERRIDE', default=None)
 """
 Optional context variable which can be used to override the default behaviour of code generation.
 Invoked with the file path to generate a download code for.

dara/core/internal/encoder_registry.py

@@ -15,14 +15,10 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from collections.abc import MutableMapping
+from collections.abc import Callable, MutableMapping
 from inspect import Parameter, isclass
 from typing import (
     Any,
-    Callable,
-    Dict,
-    Optional,
-    Type,
     Union,
     get_args,
     get_origin,
@@ -47,7 +43,7 @@ def _not_implemented(x, dtype):
     raise NotImplementedError(f'No deserialization implementation for item {x} of dtype {dtype}')
 
 
-def _get_numpy_dtypes_encoder(typ: Type[Any]):
+def _get_numpy_dtypes_encoder(typ: type[Any]):
     """
     Construct numpy generic datatype
 
@@ -56,7 +52,7 @@ def _get_numpy_dtypes_encoder(typ: Type[Any]):
     return Encoder(serialize=lambda x: x.item(), deserialize=lambda x: typ(x))
 
 
-def _get_numpy_str_encoder(typ: Type[Any]):
+def _get_numpy_str_encoder(typ: type[Any]):
     """
     Construct numpy str datatype
 
@@ -65,7 +61,7 @@ def _get_numpy_str_encoder(typ: Type[Any]):
     return Encoder(serialize=lambda x: str(x), deserialize=lambda x: typ(x))
 
 
-def _get_pandas_array_encoder(array_type: Type[Any], dtype: Any, raise_: bool = False):
+def _get_pandas_array_encoder(array_type: type[Any], dtype: Any, raise_: bool = False):
     return Encoder(
         serialize=lambda x: x.astype(str).tolist(),
         deserialize=lambda x: pandas.array(x, dtype=dtype) if not raise_ else _not_implemented(x, dtype),
@@ -127,7 +123,7 @@ def _df_deserialize(x):
 
 
 # A encoder_registry to handle serialization/deserialization for numpy/pandas type
-encoder_registry: MutableMapping[Type[Any], Encoder] = {
+encoder_registry: MutableMapping[type[Any], Encoder] = {
     int: Encoder(serialize=lambda x: x, deserialize=lambda x: int(x)),
     float: Encoder(serialize=lambda x: x, deserialize=lambda x: float(x)),
     str: Encoder(serialize=lambda x: x, deserialize=lambda x: str(x)),
@@ -211,14 +207,14 @@ else:
     )
 
 
-def get_jsonable_encoder() -> Dict[Type[Any], Callable[..., Any]]:
+def get_jsonable_encoder() -> dict[type[Any], Callable[..., Any]]:
     """
     Get the encoder registry as a dict of `{type: serialize}` pairs
     """
     return {k: v['serialize'] for k, v in encoder_registry.items()}
 
 
-def deserialize(value: Any, typ: Optional[Type]):
+def deserialize(value: Any, typ: type | None):
     """
     Deserialize a value into a given type.
 

dara/core/internal/execute_action.py

@@ -18,10 +18,10 @@ limitations under the License.
 from __future__ import annotations
 
 import asyncio
-from collections.abc import Mapping
+from collections.abc import Callable, Mapping
 from contextvars import ContextVar
 from functools import partial
-from typing import Any, Callable, Literal, Optional, Union
+from typing import Any, Literal
 
 import anyio
 
@@ -132,7 +132,7 @@ async def execute_action_sync(
     results = []
 
     # Construct a context which handles action messages by accumulating them in an array
-    async def handle_action(act_impl: Optional[ActionImpl]):
+    async def handle_action(act_impl: ActionImpl | None):
         if act_impl is not None:
             results.append(act_impl)
 
@@ -176,7 +176,7 @@ async def execute_action(
     ws_channel: str,
     store: CacheStore,
     task_mgr: TaskManager,
-) -> Union[Any, BaseTask]:
+) -> Any | BaseTask:
     """
     Execute a given action with the provided context.
 
@@ -201,7 +201,7 @@ async def execute_action(
     assert action is not None, 'Action resolver must be defined'
 
     # Construct a context which handles action messages by sending them to the frontend
-    async def handle_action(act_impl: Optional[ActionImpl]):
+    async def handle_action(act_impl: ActionImpl | None):
         await ws_mgr.send_message(ws_channel, {'action': act_impl, 'uid': execution_id})
 
     ctx = ActionCtx(inp, handle_action)

dara/core/internal/hashing.py

@@ -17,12 +17,11 @@ limitations under the License.
 
 import hashlib
 import json
-from typing import Union
 
 from pydantic import BaseModel
 
 
-def hash_object(obj: Union[BaseModel, dict, None]):
+def hash_object(obj: BaseModel | dict | None):
     """
     Create a unique hash for the object.
 

dara/core/internal/import_discovery.py

@@ -18,9 +18,7 @@ limitations under the License.
 import inspect
 import sys
 from types import ModuleType
-from typing import Any, List, Optional, Set, Tuple, Type, Union
-
-from typing_extensions import TypeGuard
+from typing import Any, TypeGuard
 
 from dara.core.base_definitions import ActionDef, ActionImpl
 from dara.core.definitions import ComponentInstance, JsComponentDef, discover
@@ -44,7 +42,7 @@ def _is_action_subclass(obj: Any) -> bool:
     return inspect.isclass(obj) and issubclass(obj, ActionImpl) and obj != ActionImpl
 
 
-def is_ignored(symbol: Any, ignore_symbols: List[Any]) -> bool:
+def is_ignored(symbol: Any, ignore_symbols: list[Any]) -> bool:
     """
     Check whether a symbol should be ignored
 
@@ -64,8 +62,8 @@ def is_ignored(symbol: Any, ignore_symbols: List[Any]) -> bool:
 
 
 def run_discovery(
-    module: Union[ModuleType, dict], ignore_symbols: Optional[List[Any]] = None, **kwargs
-) -> Tuple[Set[Type[ComponentInstance]], Set[Type[ActionImpl]]]:
+    module: ModuleType | dict, ignore_symbols: list[Any] | None = None, **kwargs
+) -> tuple[set[type[ComponentInstance]], set[type[ActionImpl]]]:
     """
     Recursively discover components available in the global namespace within the module
     and its child modules.
@@ -167,7 +165,7 @@ def run_discovery(
     return components, actions
 
 
-def _get_symbol_module(symbol: Union[Type[ComponentInstance], Type[ActionImpl]]) -> str:
+def _get_symbol_module(symbol: type[ComponentInstance] | type[ActionImpl]) -> str:
     """Get the root module of the component or action"""
     comp_module = symbol.__module__
 
@@ -182,7 +180,7 @@ def _get_symbol_module(symbol: Union[Type[ComponentInstance], Type[ActionImpl]])
     return comp_module
 
 
-def create_component_definition(component: Type[ComponentInstance], local: bool = False):
+def create_component_definition(component: type[ComponentInstance], local: bool = False):
     """
     Create a JsComponentDef for a given component class.
 
@@ -203,7 +201,7 @@ def create_component_definition(component: Type[ComponentInstance], local: bool
     )
 
 
-def create_action_definition(action: Type[ActionImpl], local: bool = False):
+def create_action_definition(action: type[ActionImpl], local: bool = False):
     """
     Create a ActionDef for a given action class.
 

dara/core/internal/normalization.py

@@ -19,20 +19,17 @@ from collections.abc import Mapping
 from typing import (
     Any,
     Generic,
-    List,
-    Optional,
-    Tuple,
+    TypeGuard,
     TypeVar,
-    Union,
     cast,
     overload,
 )
 
-from typing_extensions import TypedDict, TypeGuard
+from typing_extensions import TypedDict
 
 from dara.core.base_definitions import DaraBaseModel as BaseModel
 
-JsonLike = Union[Mapping, List]
+JsonLike = Mapping | list
 
 DataType = TypeVar('DataType')
 
@@ -60,7 +57,7 @@ class Referrable(TypedDict):
 
 
 class ReferrableWithNested(Referrable):
-    nested: List[str]
+    nested: list[str]
 
 
 class ReferrableWithFilters(Referrable):
@@ -77,7 +74,7 @@ def _get_identifier(obj: Referrable) -> str:
 
     # If it's a Variable with 'nested', the property should be included in the identifier
     if _is_referrable_nested(obj) and len(obj['nested']) > 0:
-        nested = ','.join(cast(List[str], obj['nested']))
+        nested = ','.join(cast(list[str], obj['nested']))
         identifier = f'{identifier}:{nested}'
 
     return identifier
@@ -128,14 +125,14 @@ def _loop(iterable: JsonLike):
 
 
 @overload
-def normalize(obj: Mapping, check_root: bool = True) -> Tuple[Mapping, Mapping]: ...
+def normalize(obj: Mapping, check_root: bool = True) -> tuple[Mapping, Mapping]: ...
 
 
 @overload
-def normalize(obj: List, check_root: bool = True) -> Tuple[List, Mapping]: ...
+def normalize(obj: list, check_root: bool = True) -> tuple[list, Mapping]: ...
 
 
-def normalize(obj: JsonLike, check_root: bool = True) -> Tuple[JsonLike, Mapping]:
+def normalize(obj: JsonLike, check_root: bool = True) -> tuple[JsonLike, Mapping]:
     """
     Normalize a dictionary - extract referrable data into a separate lookup dictionary, replacing instances
     found with placeholders.
@@ -148,7 +145,7 @@ def normalize(obj: JsonLike, check_root: bool = True) -> Tuple[JsonLike, Mapping
     if not isinstance(obj, (dict, list)):
         return obj, lookup
 
-    output: Union[Mapping[Any, Any], List[Any]] = {} if isinstance(obj, dict) else [None for x in range(len(obj))]
+    output: Mapping[Any, Any] | list[Any] = {} if isinstance(obj, dict) else [None for x in range(len(obj))]
 
     # The whole object is referrable
     if check_root and _is_referrable(obj):
@@ -177,10 +174,10 @@ def denormalize(normalized_obj: Mapping, lookup: Mapping) -> Mapping: ...
 
 
 @overload
-def denormalize(normalized_obj: List, lookup: Mapping) -> List: ...
+def denormalize(normalized_obj: list, lookup: Mapping) -> list: ...
 
 
-def denormalize(normalized_obj: JsonLike, lookup: Mapping) -> Optional[JsonLike]:
+def denormalize(normalized_obj: JsonLike, lookup: Mapping) -> JsonLike | None:
     """
     Denormalize data by replacing Placeholders found with objects from the lookup
 
@@ -190,7 +187,7 @@ def denormalize(normalized_obj: JsonLike, lookup: Mapping) -> Optional[JsonLike]
     if normalized_obj is None:
         return None
 
-    output: Union[Mapping[Any, Any], List[Any]] = (
+    output: Mapping[Any, Any] | list[Any] = (
         {} if isinstance(normalized_obj, dict) else [None for x in range(len(normalized_obj))]
     )
 

dara/core/internal/pandas_utils.py

@@ -17,10 +17,10 @@ limitations under the License.
 
 import json
 import uuid
-from typing import Any, Literal, Optional, TypeVar, Union, cast, overload
+from typing import Any, Literal, TypeGuard, TypeVar, cast, overload
 
 from pandas import DataFrame, MultiIndex, Series
-from typing_extensions import TypedDict, TypeGuard
+from typing_extensions import TypedDict
 
 INDEX = '__index__'
 
@@ -33,7 +33,7 @@ def append_index(df: DataFrame) -> DataFrame: ...
 def append_index(df: None) -> None: ...
 
 
-def append_index(df: Optional[DataFrame]) -> Optional[DataFrame]:
+def append_index(df: DataFrame | None) -> DataFrame | None:
     """
     Add a numerical index column to the dataframe
     """
@@ -130,7 +130,7 @@ def format_for_display(df: DataFrame) -> None:
 
 
 class FieldType(TypedDict):
-    name: Union[str, tuple[str, ...]]
+    name: str | tuple[str, ...]
     type: Literal['integer', 'number', 'boolean', 'datetime', 'duration', 'any', 'str']
 
 
@@ -140,9 +140,9 @@ class DataFrameSchema(TypedDict):
 
 
 class DataResponse(TypedDict):
-    data: Optional[DataFrame]
+    data: DataFrame | None
     count: int
-    schema: Optional[DataFrameSchema]
+    schema: DataFrameSchema | None
 
 
 def is_data_response(response: Any) -> TypeGuard[DataResponse]:

dara/core/internal/pool/channel.py

@@ -20,7 +20,6 @@ from __future__ import annotations
 import os
 from multiprocessing import Queue, get_context
 from queue import Empty
-from typing import Optional
 
 from dara.core.internal.pool.definitions import (
     Acknowledgement,
@@ -56,7 +55,7 @@ class _PoolAPI:
 
     def get_worker_message(
         self,
-    ) -> Optional[WorkerMessage]:
+    ) -> WorkerMessage | None:
         """
         Retrieve a worker message if there is one available
 
@@ -100,7 +99,7 @@ class _WorkerAPI:
         """
         self._out_queue.put(Result(task_uid=task_uid, result=result))
 
-    def send_error(self, task_uid: Optional[str], error: BaseException):
+    def send_error(self, task_uid: str | None, error: BaseException):
         """
         Send an error back to the pool
 
@@ -131,7 +130,7 @@ class _WorkerAPI:
         """
         self._out_queue.put(Progress(task_uid=task_uid, progress=progress, message=message))
 
-    def get_task(self) -> Optional[WorkerTask]:
+    def get_task(self) -> WorkerTask | None:
         """
         Retrieve a task definition from the worker queue if there is one available
 

dara/core/internal/pool/definitions.py

@@ -17,10 +17,10 @@ limitations under the License.
 
 from datetime import datetime
 from enum import Enum
-from typing import Any, Optional, Union
+from typing import Any, TypeGuard
 
 from anyio import Event
-from typing_extensions import TypedDict, TypeGuard
+from typing_extensions import TypedDict
 
 from dara.core.internal.pool.utils import SharedMemoryPointer, SubprocessException
 
@@ -75,16 +75,16 @@ class TaskDefinition:
     event: Event
     result: Any
     payload: TaskPayload
-    worker_id: Optional[int] = None
-    started_at: Optional[datetime] = None
+    worker_id: int | None = None
+    started_at: datetime | None = None
     """TODO: can be used for task timeout or metrics/visibility"""
 
     def __init__(
         self,
         uid: str,
         payload: TaskPayload,
-        worker_id: Optional[int] = None,
-        started_at: Optional[datetime] = None,
+        worker_id: int | None = None,
+        started_at: datetime | None = None,
     ):
         self.uid = uid
         self.payload = payload
@@ -135,14 +135,14 @@ class Result(TypedDict):
 class Problem(TypedDict):
     """Sent when a worker encounters an issue processing a task"""
 
-    task_uid: Optional[str]
+    task_uid: str | None
     error: SubprocessException
 
 
 class Log(TypedDict):
     """Sent when a task emits a stdout message"""
 
-    task_uid: Optional[str]
+    task_uid: str | None
     log: str
 
 
@@ -154,7 +154,7 @@ class Progress(TypedDict):
     message: str
 
 
-WorkerMessage = Union[Acknowledgement, Result, Problem, Initialization, Log, Progress]
+WorkerMessage = Acknowledgement | Result | Problem | Initialization | Log | Progress
 """Union of possible messages sent from worker processes"""
 
 

dara/core/internal/pool/task_pool.py

@@ -16,11 +16,11 @@ limitations under the License.
 """
 
 import atexit
-from collections.abc import Coroutine
+from collections.abc import Callable, Coroutine
 from contextlib import contextmanager
 from datetime import datetime
 from multiprocessing import active_children
-from typing import Any, Callable, Dict, Optional, Union, cast
+from typing import Any, cast
 
 import anyio
 from anyio.abc import TaskGroup
@@ -61,8 +61,8 @@ class TaskPool:
     """Number of seconds worker is allowed to be idle before it is killed, if there are too many workers alive"""
 
     worker_parameters: WorkerParameters
-    workers: Dict[int, WorkerProcess] = {}
-    tasks: Dict[str, TaskDefinition] = {}
+    workers: dict[int, WorkerProcess] = {}
+    tasks: dict[str, TaskDefinition] = {}
 
     def __init__(
         self, task_group: TaskGroup, worker_parameters: WorkerParameters, max_workers: int, worker_timeout: float = 5
@@ -75,7 +75,7 @@ class TaskPool:
         self.worker_timeout = worker_timeout
 
         self._channel = Channel()
-        self._progress_subscribers: Dict[str, Callable[[float, str], Coroutine]] = {}
+        self._progress_subscribers: dict[str, Callable[[float, str], Coroutine]] = {}
 
     @property
     def running_tasks(self):
@@ -113,7 +113,7 @@ class TaskPool:
             raise RuntimeError('Pool already started')
 
     def submit(
-        self, task_uid: str, function_name: str, args: Union[tuple, None] = None, kwargs: Union[dict, None] = None
+        self, task_uid: str, function_name: str, args: tuple | None = None, kwargs: dict | None = None
     ) -> TaskDefinition:
         """
         Submit a new task to the pool
@@ -196,7 +196,7 @@ class TaskPool:
         await self.loop_stopped.wait()
         await self._terminate_workers()
 
-    async def join(self, timeout: Optional[float] = None):
+    async def join(self, timeout: float | None = None):
         """
         Join the pool and wait for workers to complete
 
@@ -473,7 +473,7 @@ class TaskPool:
         elif is_progress(worker_msg) and worker_msg['task_uid'] in self._progress_subscribers:
             await self._progress_subscribers[worker_msg['task_uid']](worker_msg['progress'], worker_msg['message'])
 
-    async def _wait_queue_depletion(self, timeout: Optional[float] = None):
+    async def _wait_queue_depletion(self, timeout: float | None = None):
         """
         Wait until all tasks have been marked as completed
 

dara/core/internal/pool/utils.py

@@ -20,9 +20,10 @@ import os
 import pickle
 import signal
 import sys
+from collections.abc import Callable
 from multiprocessing.process import BaseProcess
 from multiprocessing.shared_memory import SharedMemory
-from typing import Any, Callable, Optional, Tuple
+from typing import Any
 
 import anyio
 from tblib import Traceback
@@ -48,7 +49,7 @@ class SubprocessException:
         return self.exception.with_traceback(tb)
 
 
-SharedMemoryPointer = Tuple[str, int]
+SharedMemoryPointer = tuple[str, int]
 
 
 class PicklingException(Exception):
@@ -106,7 +107,7 @@ def read_from_shared_memory(pointer: SharedMemoryPointer) -> Any:
         raise PicklingException(*e.args) from e
 
 
-async def wait_while(condition: Callable[[], bool], timeout: Optional[float] = None):
+async def wait_while(condition: Callable[[], bool], timeout: float | None = None):
     """
     Util to wait until a condition is False or timeout is exceeded
 

dara/core/internal/pool/worker.py

@@ -21,13 +21,13 @@ import logging
 import os
 import signal
 import sys
+from collections.abc import Callable
 from datetime import datetime
 from importlib import import_module
 from inspect import iscoroutinefunction
 from multiprocessing import get_context
 from multiprocessing.context import SpawnProcess
 from time import sleep
-from typing import Callable, Optional
 
 import anyio
 
@@ -195,7 +195,7 @@ class WorkerProcess:
 
     status: WorkerStatus
 
-    task_uid: Optional[str] = None
+    task_uid: str | None = None
     """Current task UID being processed by the worker"""
 
     channel: Channel
@@ -213,7 +213,7 @@ class WorkerProcess:
         self.process = ctx.Process(target=worker_loop, args=(worker_params, channel), name=WORKER_NAME)
         self.process.start()
 
-    def update_status(self, worker_status: WorkerStatus, task_uid: Optional[str] = None):
+    def update_status(self, worker_status: WorkerStatus, task_uid: str | None = None):
         self.status = worker_status
         self.updated_at = datetime.now()
         self.task_uid = task_uid

dara/core/internal/registries.py

@@ -15,9 +15,9 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from collections.abc import Mapping
+from collections.abc import Callable, Mapping
 from datetime import datetime
-from typing import Any, Callable, Set
+from typing import Any
 
 from dara.core.auth import BaseAuthConfig
 from dara.core.base_definitions import ActionDef, ActionResolverDef, UploadResolverDef
@@ -55,10 +55,10 @@ auth_registry = Registry[BaseAuthConfig](RegistryType.AUTH_CONFIG)
 utils_registry = Registry[Any](RegistryType.UTILS, INITIAL_CORE_INTERNALS)
 static_kwargs_registry = Registry[Mapping[str, Any]](RegistryType.STATIC_KWARGS)
 
-websocket_registry = Registry[Set[str]](RegistryType.WEBSOCKET_CHANNELS)
+websocket_registry = Registry[set[str]](RegistryType.WEBSOCKET_CHANNELS)
 """maps session_id -> WS channel"""
 
-sessions_registry = Registry[Set[str]](RegistryType.USER_SESSION)
+sessions_registry = Registry[set[str]](RegistryType.USER_SESSION)
 """maps user_identifier -> session_ids """
 
 pending_tokens_registry = Registry[datetime](RegistryType.PENDING_TOKENS)