diff --git a/python/aqora/__init__.py b/python/aqora/__init__.py index 7147fd4..ea4a268 100644 --- a/python/aqora/__init__.py +++ b/python/aqora/__init__.py @@ -1,7 +1,7 @@ from typing_extensions import Any, TypedDict, override from . import _aqora -from .notebook import notebook +from .notebook import load from aqora._aqora import * # pyright: ignore[reportAssignmentType, reportWildcardImportFromLibrary] # noqa: F403 diff --git a/python/aqora/notebook.py b/python/aqora/notebook.py index a0902a5..ed50976 100644 --- a/python/aqora/notebook.py +++ b/python/aqora/notebook.py @@ -96,7 +96,7 @@ def _load_module(owner: str, slug: str, filename: str, path: Path) -> ModuleType return module -def notebook( +def load( workspace: str, *, filename: str | None = None,