diff --git a/python/aqora_cli/__init__.py b/python/aqora_cli/__init__.py index e4f5eaf..eaac4fe 100644 --- a/python/aqora_cli/__init__.py +++ b/python/aqora_cli/__init__.py @@ -1,7 +1,7 @@ from typing_extensions import Any, TypedDict, override from . import _aqora_cli -from .notebook import notebook +from .notebook import load from aqora_cli._aqora_cli import * # pyright: ignore[reportAssignmentType, reportWildcardImportFromLibrary] # noqa: F403 diff --git a/python/aqora_cli/notebook.py b/python/aqora_cli/notebook.py index 2c569c2..6056ef7 100644 --- a/python/aqora_cli/notebook.py +++ b/python/aqora_cli/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,