streamlit-nightly 1.31.2.dev20240212__py2.py3-none-any.whl → 1.31.2.dev20240214__py2.py3-none-any.whl

streamlit/runtime/app_session.py

@@ -11,11 +11,14 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from __future__ import annotations
+
 import asyncio
 import sys
 import uuid
 from enum import Enum
-from typing import TYPE_CHECKING, Callable, Dict, List, Optional, Union
+from typing import TYPE_CHECKING, Callable, Final
 
 import streamlit.elements.exception as exception_utils
 from streamlit import config, runtime, source_util
@@ -44,10 +47,11 @@ from streamlit.runtime.uploaded_file_manager import UploadedFileManager
 from streamlit.version import STREAMLIT_VERSION_STRING
 from streamlit.watcher import LocalSourcesWatcher
 
-LOGGER = get_logger(__name__)
 if TYPE_CHECKING:
     from streamlit.runtime.state import SessionState
 
+_LOGGER: Final = get_logger(__name__)
+
 
 class AppSessionState(Enum):
     APP_NOT_RUNNING = "APP_NOT_RUNNING"
@@ -77,10 +81,10 @@ class AppSession:
         script_data: ScriptData,
         uploaded_file_manager: UploadedFileManager,
         script_cache: ScriptCache,
-        message_enqueued_callback: Optional[Callable[[], None]],
+        message_enqueued_callback: Callable[[], None] | None,
         local_sources_watcher: LocalSourcesWatcher,
-        user_info: Dict[str, Optional[str]],
-        session_id_override: Optional[str] = None,
+        user_info: dict[str, str | None],
+        session_id_override: str | None = None,
     ) -> None:
         """Initialize the AppSession.
 
@@ -139,17 +143,15 @@ class AppSession:
         # due to the source code changing we need to pass in the previous client state.
         self._client_state = ClientState()
 
-        self._local_sources_watcher: Optional[
-            LocalSourcesWatcher
-        ] = local_sources_watcher
-        self._stop_config_listener: Optional[Callable[[], bool]] = None
-        self._stop_pages_listener: Optional[Callable[[], bool]] = None
+        self._local_sources_watcher: LocalSourcesWatcher | None = local_sources_watcher
+        self._stop_config_listener: Callable[[], bool] | None = None
+        self._stop_pages_listener: Callable[[], None] | None = None
 
         self.register_file_watchers()
 
         self._run_on_save = config.get_option("server.runOnSave")
 
-        self._scriptrunner: Optional[ScriptRunner] = None
+        self._scriptrunner: ScriptRunner | None = None
 
         # This needs to be lazily imported to avoid a dependency cycle.
         from streamlit.runtime.state import SessionState
@@ -157,9 +159,9 @@ class AppSession:
         self._session_state = SessionState()
         self._user_info = user_info
 
-        self._debug_last_backmsg_id: Optional[str] = None
+        self._debug_last_backmsg_id: str | None = None
 
-        LOGGER.debug("AppSession initialized (id=%s)", self.id)
+        _LOGGER.debug("AppSession initialized (id=%s)", self.id)
 
     def __del__(self) -> None:
         """Ensure that we call shutdown() when an AppSession is garbage collected."""
@@ -210,7 +212,7 @@ class AppSession:
         self._stop_config_listener = None
         self._stop_pages_listener = None
 
-    def flush_browser_queue(self) -> List[ForwardMsg]:
+    def flush_browser_queue(self) -> list[ForwardMsg]:
         """Clear the forward message queue and return the messages it contained.
 
         The Server calls this periodically to deliver new messages
@@ -232,7 +234,7 @@ class AppSession:
 
         """
         if self._state != AppSessionState.SHUTDOWN_REQUESTED:
-            LOGGER.debug("Shutting down (id=%s)", self.id)
+            _LOGGER.debug("Shutting down (id=%s)", self.id)
             # Clear any unused session files in upload file manager and media
             # file manager
             self._uploaded_file_mgr.remove_session_files(self.id)
@@ -298,10 +300,10 @@ class AppSession:
             elif msg_type == "file_urls_request":
                 self._handle_file_urls_request(msg.file_urls_request)
             else:
-                LOGGER.warning('No handler for "%s"', msg_type)
+                _LOGGER.warning('No handler for "%s"', msg_type)
 
         except Exception as ex:
-            LOGGER.error(ex)
+            _LOGGER.error(ex)
             self.handle_backmsg_exception(ex)
 
     def handle_backmsg_exception(self, e: BaseException) -> None:
@@ -333,7 +335,7 @@ class AppSession:
             lambda: self._enqueue_forward_msg(self._create_exception_message(e))
         )
 
-    def request_rerun(self, client_state: Optional[ClientState]) -> None:
+    def request_rerun(self, client_state: ClientState | None) -> None:
         """Signal that we're interested in running the script.
 
         If the script is not already running, it will be started immediately.
@@ -347,7 +349,7 @@ class AppSession:
 
         """
         if self._state == AppSessionState.SHUTDOWN_REQUESTED:
-            LOGGER.warning("Discarding rerun request after shutdown")
+            _LOGGER.warning("Discarding rerun request after shutdown")
             return
 
         if client_state:
@@ -403,7 +405,7 @@ class AppSession:
         self._scriptrunner.start()
 
     @property
-    def session_state(self) -> "SessionState":
+    def session_state(self) -> SessionState:
         return self._session_state
 
     def _should_rerun_on_file_change(self, filepath: str) -> bool:
@@ -421,7 +423,7 @@ class AppSession:
 
         return True
 
-    def _on_source_file_changed(self, filepath: Optional[str] = None) -> None:
+    def _on_source_file_changed(self, filepath: str | None = None) -> None:
         """One of our source files changed. Clear the cache and schedule a rerun if appropriate."""
         self._script_cache.clear()
 
@@ -456,12 +458,12 @@ class AppSession:
 
     def _on_scriptrunner_event(
         self,
-        sender: Optional[ScriptRunner],
+        sender: ScriptRunner | None,
         event: ScriptRunnerEvent,
-        forward_msg: Optional[ForwardMsg] = None,
-        exception: Optional[BaseException] = None,
-        client_state: Optional[ClientState] = None,
-        page_script_hash: Optional[str] = None,
+        forward_msg: ForwardMsg | None = None,
+        exception: BaseException | None = None,
+        client_state: ClientState | None = None,
+        page_script_hash: str | None = None,
     ) -> None:
         """Called when our ScriptRunner emits an event.
 
@@ -477,12 +479,12 @@ class AppSession:
 
     def _handle_scriptrunner_event_on_event_loop(
         self,
-        sender: Optional[ScriptRunner],
+        sender: ScriptRunner | None,
         event: ScriptRunnerEvent,
-        forward_msg: Optional[ForwardMsg] = None,
-        exception: Optional[BaseException] = None,
-        client_state: Optional[ClientState] = None,
-        page_script_hash: Optional[str] = None,
+        forward_msg: ForwardMsg | None = None,
+        exception: BaseException | None = None,
+        client_state: ClientState | None = None,
+        page_script_hash: str | None = None,
     ) -> None:
         """Handle a ScriptRunner event.
 
@@ -525,7 +527,7 @@ class AppSession:
             # rerun request, for example) while another ScriptRunner is still
             # shutting down. The shutting-down ScriptRunner may still
             # emit events.
-            LOGGER.debug("Ignoring event from non-current ScriptRunner: %s", event)
+            _LOGGER.debug("Ignoring event from non-current ScriptRunner: %s", event)
             return
 
         prev_state = self._state
@@ -663,7 +665,7 @@ class AppSession:
         return msg
 
     def _create_script_finished_message(
-        self, status: "ForwardMsg.ScriptFinishedStatus.ValueType"
+        self, status: ForwardMsg.ScriptFinishedStatus.ValueType
     ) -> ForwardMsg:
         """Create and return a script_finished ForwardMsg."""
         msg = ForwardMsg()
@@ -712,10 +714,10 @@ class AppSession:
         except Exception as ex:
             # Users may never even install Git in the first place, so this
             # error requires no action. It can be useful for debugging.
-            LOGGER.debug("Obtaining Git information produced an error", exc_info=ex)
+            _LOGGER.debug("Obtaining Git information produced an error", exc_info=ex)
 
     def _handle_rerun_script_request(
-        self, client_state: Optional[ClientState] = None
+        self, client_state: ClientState | None = None
     ) -> None:
         """Tell the ScriptRunner to re-run its script.
 
@@ -794,10 +796,10 @@ class AppSession:
 # This field will be available at runtime as of protobuf 3.20.1, but
 # we are using an older version.
 # For details, see: https://github.com/protocolbuffers/protobuf/issues/8175
-def _get_toolbar_mode() -> "Config.ToolbarMode.ValueType":
+def _get_toolbar_mode() -> Config.ToolbarMode.ValueType:
     config_key = "client.toolbarMode"
     config_value = config.get_option(config_key)
-    enum_value: Optional["Config.ToolbarMode.ValueType"] = getattr(
+    enum_value: Config.ToolbarMode.ValueType | None = getattr(
         Config.ToolbarMode, config_value.upper()
     )
     if enum_value is None:
@@ -844,7 +846,7 @@ def _populate_theme_msg(msg: CustomThemeConfig) -> None:
     base = theme_opts["base"]
     if base is not None:
         if base not in base_map:
-            LOGGER.warning(
+            _LOGGER.warning(
                 f'"{base}" is an invalid value for theme.base.'
                 f" Allowed values include {list(base_map.keys())}."
                 ' Setting theme.base to "light".'
@@ -860,7 +862,7 @@ def _populate_theme_msg(msg: CustomThemeConfig) -> None:
     font = theme_opts["font"]
     if font is not None:
         if font not in font_map:
-            LOGGER.warning(
+            _LOGGER.warning(
                 f'"{font}" is an invalid value for theme.font.'
                 f" Allowed values include {list(font_map.keys())}."
                 ' Setting theme.font to "sans serif".'
@@ -874,9 +876,7 @@ def _populate_user_info_msg(msg: UserInfo) -> None:
     msg.installation_id_v3 = Installation.instance().installation_id_v3
 
 
-def _populate_app_pages(
-    msg: Union[NewSession, PagesChanged], main_script_path: str
-) -> None:
+def _populate_app_pages(msg: NewSession | PagesChanged, main_script_path: str) -> None:
     for page_script_hash, page_info in source_util.get_pages(main_script_path).items():
         page_proto = msg.app_pages.add()
 

streamlit/runtime/caching/__init__.py

@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import contextlib
-from typing import Any, Iterator, Union
+from typing import Any, Iterator
 
 from google.protobuf.message import Message
 
@@ -77,9 +79,7 @@ def save_widget_metadata(metadata: WidgetMetadata[Any]) -> None:
     CACHE_RESOURCE_MESSAGE_REPLAY_CTX.save_widget_metadata(metadata)
 
 
-def save_media_data(
-    image_data: Union[bytes, str], mimetype: str, image_id: str
-) -> None:
+def save_media_data(image_data: bytes | str, mimetype: str, image_id: str) -> None:
     CACHE_DATA_MESSAGE_REPLAY_CTX.save_image_data(image_data, mimetype, image_id)
     CACHE_RESOURCE_MESSAGE_REPLAY_CTX.save_image_data(image_data, mimetype, image_id)
 

streamlit/runtime/caching/cache_data_api.py

@@ -20,9 +20,9 @@ import pickle
 import threading
 import types
 from datetime import timedelta
-from typing import Any, Callable, TypeVar, Union, cast, overload
+from typing import Any, Callable, Final, Literal, TypeVar, Union, cast, overload
 
-from typing_extensions import Literal, TypeAlias
+from typing_extensions import TypeAlias
 
 import streamlit as st
 from streamlit import runtime
@@ -62,7 +62,7 @@ from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner.script_run_context import get_script_run_ctx
 from streamlit.runtime.stats import CacheStat, CacheStatsProvider, group_stats
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 CACHE_DATA_MESSAGE_REPLAY_CTX = CachedMessageReplayContext(CacheType.DATA)
 

streamlit/runtime/caching/cache_errors.py

@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import types
-from typing import Any, Optional
+from typing import Any
 
 from streamlit import type_util
 from streamlit.errors import (
@@ -50,7 +52,7 @@ class UnhashableParamError(StreamlitAPIException):
         self,
         cache_type: CacheType,
         func: types.FunctionType,
-        arg_name: Optional[str],
+        arg_name: str | None,
         arg_value: Any,
         orig_exc: BaseException,
     ):
@@ -62,7 +64,7 @@ class UnhashableParamError(StreamlitAPIException):
     def _create_message(
         cache_type: CacheType,
         func: types.FunctionType,
-        arg_name: Optional[str],
+        arg_name: str | None,
         arg_value: Any,
     ) -> str:
         arg_name_str = arg_name if arg_name is not None else "(unnamed)"

streamlit/runtime/caching/cache_type.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import enum
 
 

streamlit/runtime/caching/cache_utils.py

@@ -26,9 +26,7 @@ import types
 from abc import abstractmethod
 from collections import defaultdict
 from datetime import timedelta
-from typing import Any, Callable, overload
-
-from typing_extensions import Literal
+from typing import Any, Callable, Final, Literal, overload
 
 from streamlit import type_util
 from streamlit.elements.spinner import spinner
@@ -53,7 +51,7 @@ from streamlit.runtime.caching.cached_message_replay import (
 from streamlit.runtime.caching.hashing import HashFuncsDict, update_hash
 from streamlit.util import HASHLIB_KWARGS
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 # The timer function we use with TTLCache. This is the default timer func, but
 # is exposed here as a constant so that it can be patched in unit tests.

streamlit/runtime/caching/cached_message_replay.py

@@ -19,10 +19,17 @@ import hashlib
 import threading
 import types
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any, Iterator, Union
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Final,
+    Iterator,
+    Protocol,
+    Union,
+    runtime_checkable,
+)
 
 from google.protobuf.message import Message
-from typing_extensions import Protocol, runtime_checkable
 
 import streamlit as st
 from streamlit import runtime, util
@@ -45,7 +52,7 @@ from streamlit.util import HASHLIB_KWARGS
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 
 @runtime_checkable
@@ -369,7 +376,7 @@ class CachedMessageReplayContext(threading.local):
 
     def maybe_show_cached_st_function_warning(
         self,
-        dg: "DeltaGenerator",
+        dg: DeltaGenerator,
         st_func_name: str,
     ) -> None:
         """If appropriate, warn about calling st.foo inside @memo.
@@ -401,7 +408,7 @@ class CachedMessageReplayContext(threading.local):
 
     def _show_cached_st_function_warning(
         self,
-        dg: "DeltaGenerator",
+        dg: DeltaGenerator,
         st_func_name: str,
         cached_func: types.FunctionType,
     ) -> None:

streamlit/runtime/caching/hashing.py

@@ -13,6 +13,9 @@
 # limitations under the License.
 
 """Hashing for st.cache_data and st.cache_resource."""
+
+from __future__ import annotations
+
 import collections
 import dataclasses
 import datetime
@@ -22,15 +25,15 @@ import inspect
 import io
 import os
 import pickle
-import struct
 import sys
 import tempfile
 import threading
-import unittest.mock
 import uuid
 import weakref
 from enum import Enum
-from typing import Any, Callable, Dict, List, Optional, Pattern, Type, Union
+from typing import Any, Callable, Dict, Final, Pattern, Type, Union
+
+from typing_extensions import TypeAlias
 
 from streamlit import type_util, util
 from streamlit.errors import StreamlitAPIException
@@ -40,18 +43,20 @@ from streamlit.runtime.uploaded_file_manager import UploadedFile
 from streamlit.util import HASHLIB_KWARGS
 
 # If a dataframe has more than this many rows, we consider it large and hash a sample.
-_PANDAS_ROWS_LARGE = 100000
-_PANDAS_SAMPLE_SIZE = 10000
+_PANDAS_ROWS_LARGE: Final = 100000
+_PANDAS_SAMPLE_SIZE: Final = 10000
 
 # Similar to dataframes, we also sample large numpy arrays.
-_NP_SIZE_LARGE = 1000000
-_NP_SAMPLE_SIZE = 100000
+_NP_SIZE_LARGE: Final = 1000000
+_NP_SAMPLE_SIZE: Final = 100000
 
-HashFuncsDict = Dict[Union[str, Type[Any]], Callable[[Any], Any]]
+HashFuncsDict: TypeAlias = Dict[Union[str, Type[Any]], Callable[[Any], Any]]
 
 # Arbitrary item to denote where we found a cycle in a hashed object.
 # This allows us to hash self-referencing lists, dictionaries, etc.
-_CYCLE_PLACEHOLDER = b"streamlit-57R34ML17-hesamagicalponyflyingthroughthesky-CYCLE"
+_CYCLE_PLACEHOLDER: Final = (
+    b"streamlit-57R34ML17-hesamagicalponyflyingthroughthesky-CYCLE"
+)
 
 
 class UserHashError(StreamlitAPIException):
@@ -60,7 +65,7 @@ class UserHashError(StreamlitAPIException):
         orig_exc,
         object_to_hash,
         hash_func,
-        cache_type: Optional[CacheType] = None,
+        cache_type: CacheType | None = None,
     ):
         self.alternate_name = type(orig_exc).__name__
         self.hash_func = hash_func
@@ -101,7 +106,7 @@ If you think this is actually a Streamlit bug, please
         self,
         orig_exc: BaseException,
         failed_obj: Any,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         hash_source = hash_stacks.current.hash_source
 
         failed_obj_type_str = type_util.get_fqn_type(failed_obj)
@@ -139,8 +144,8 @@ def update_hash(
     val: Any,
     hasher,
     cache_type: CacheType,
-    hash_source: Optional[Callable[..., Any]] = None,
-    hash_funcs: Optional[HashFuncsDict] = None,
+    hash_source: Callable[..., Any] | None = None,
+    hash_funcs: HashFuncsDict | None = None,
 ) -> None:
     """Updates a hashlib hasher with the hash of val.
 
@@ -166,10 +171,10 @@ class _HashStack:
     """
 
     def __init__(self):
-        self._stack: collections.OrderedDict[int, List[Any]] = collections.OrderedDict()
+        self._stack: collections.OrderedDict[int, list[Any]] = collections.OrderedDict()
         # A function that we decorate with streamlit cache
         # primitive (st.cache_data or st.cache_resource).
-        self.hash_source: Optional[Callable[..., Any]] = None
+        self.hash_source: Callable[..., Any] | None = None
 
     def __repr__(self) -> str:
         return util.repr_(self)
@@ -226,11 +231,14 @@ def _int_to_bytes(i: int) -> bytes:
 
 
 def _float_to_bytes(f: float) -> bytes:
+    # Lazy-load for performance reasons.
+    import struct
+
     # Floats are 64bit in Python, so we need to use the "d" format.
     return struct.pack("<d", f)
 
 
-def _key(obj: Optional[Any]) -> Any:
+def _key(obj: Any | None) -> Any:
     """Return key for memoization."""
 
     if obj is None:
@@ -268,9 +276,7 @@ def _key(obj: Optional[Any]) -> Any:
 class _CacheFuncHasher:
     """A hasher that can hash objects with cycles."""
 
-    def __init__(
-        self, cache_type: CacheType, hash_funcs: Optional[HashFuncsDict] = None
-    ):
+    def __init__(self, cache_type: CacheType, hash_funcs: HashFuncsDict | None = None):
         # Can't use types as the keys in the internal _hash_funcs because
         # we always remove user-written modules from memory when rerunning a
         # script in order to reload it and grab the latest code changes.
@@ -286,7 +292,7 @@ class _CacheFuncHasher:
             }
         else:
             self._hash_funcs = {}
-        self._hashes: Dict[Any, bytes] = {}
+        self._hashes: dict[Any, bytes] = {}
 
         # The number of the bytes in the hash.
         self.size = 0
@@ -344,7 +350,9 @@ class _CacheFuncHasher:
 
         h = hashlib.new("md5", **HASHLIB_KWARGS)
 
-        if isinstance(obj, unittest.mock.Mock):
+        if type_util.is_type(obj, "unittest.mock.Mock") or type_util.is_type(
+            obj, "unittest.mock.MagicMock"
+        ):
             # Mock objects can appear to be infinitely
             # deep, so we don't try to hash them at all.
             return self.to_bytes(id(obj))

streamlit/runtime/caching/storage/cache_storage_protocol.py

@@ -57,8 +57,7 @@ from __future__ import annotations
 
 from abc import abstractmethod
 from dataclasses import dataclass
-
-from typing_extensions import Literal, Protocol
+from typing import Literal, Protocol
 
 
 class CacheStorageError(Exception):

streamlit/runtime/caching/storage/local_disk_cache_storage.py

@@ -61,6 +61,7 @@ from __future__ import annotations
 import math
 import os
 import shutil
+from typing import Final
 
 from streamlit import util
 from streamlit.file_util import get_streamlit_file_path, streamlit_read, streamlit_write
@@ -76,16 +77,16 @@ from streamlit.runtime.caching.storage.in_memory_cache_storage_wrapper import (
     InMemoryCacheStorageWrapper,
 )
 
+_LOGGER: Final = get_logger(__name__)
+
 # Streamlit directory where persisted @st.cache_data objects live.
 # (This is the same directory that @st.cache persisted objects live.
 # But @st.cache_data uses a different extension, so they don't overlap.)
-_CACHE_DIR_NAME = "cache"
+_CACHE_DIR_NAME: Final = "cache"
 
 # The extension for our persisted @st.cache_data objects.
 # (`@st.cache_data` was originally called `@st.memo`)
-_CACHED_FILE_EXTENSION = "memo"
-
-_LOGGER = get_logger(__name__)
+_CACHED_FILE_EXTENSION: Final = "memo"
 
 
 class LocalDiskCacheStorageManager(CacheStorageManager):
@@ -168,7 +169,7 @@ class LocalDiskCacheStorage(CacheStorage):
                 # Clean up file so we don't leave zero byte files.
                 try:
                     os.remove(path)
-                except (FileNotFoundError, IOError, OSError):
+                except (FileNotFoundError, OSError):
                     # If we can't remove the file, it's not a big deal.
                     pass
                 raise CacheStorageError("Unable to write to cache") from e

streamlit/runtime/connection_factory.py

@@ -14,13 +14,10 @@
 
 from __future__ import annotations
 
-import importlib
 import os
 import re
 from datetime import timedelta
-from typing import Any, Dict, Type, TypeVar, overload
-
-from typing_extensions import Final, Literal
+from typing import Any, Final, Literal, TypeVar, overload
 
 from streamlit.connections import (
     BaseConnection,
@@ -45,7 +42,7 @@ FIRST_PARTY_CONNECTIONS = {
     "sql": SQLConnection,
 }
 MODULE_EXTRACTION_REGEX = re.compile(r"No module named \'(.+)\'")
-MODULES_TO_PYPI_PACKAGES: Final[Dict[str, str]] = {
+MODULES_TO_PYPI_PACKAGES: Final[dict[str, str]] = {
     "MySQLdb": "mysqlclient",
     "psycopg2": "psycopg2-binary",
     "sqlalchemy": "sqlalchemy",
@@ -63,7 +60,7 @@ ConnectionClass = TypeVar("ConnectionClass", bound=BaseConnection[Any])
 @gather_metrics("connection")
 def _create_connection(
     name: str,
-    connection_class: Type[ConnectionClass],
+    connection_class: type[ConnectionClass],
     max_entries: int | None = None,
     ttl: float | timedelta | None = None,
     **kwargs,
@@ -77,7 +74,7 @@ def _create_connection(
     """
 
     def __create_connection(
-        name: str, connection_class: Type[ConnectionClass], **kwargs
+        name: str, connection_class: type[ConnectionClass], **kwargs
     ) -> ConnectionClass:
         return connection_class(connection_name=name, **kwargs)
 
@@ -184,7 +181,7 @@ def connection_factory(
 @overload
 def connection_factory(
     name: str,
-    type: Type[ConnectionClass],
+    type: type[ConnectionClass],
     max_entries: int | None = None,
     ttl: float | timedelta | None = None,
     **kwargs,
@@ -312,6 +309,9 @@ def connection_factory(
         if "." in connection_class:
             parts = connection_class.split(".")
             classname = parts.pop()
+
+            import importlib
+
             connection_module = importlib.import_module(".".join(parts))
             connection_class = getattr(connection_module, classname)
         else: