ci: move from aqora-cli to aqora for pypi

pyproject.toml

@@ -3,7 +3,7 @@ requires = ["maturin>=1.4,<2.0"]
 build-backend = "maturin"
 
 [project]
-name = "aqora-cli"
+name = "aqora"
 description = "The aqora command line interface"
 authors = [{ name = "S.A.S Aqora Quantum", email = "hello@aqora.io" }]
 requires-python = ">=3.8"
@@ -14,7 +14,7 @@ dynamic = ["version"]
 dependencies = ["typing-extensions>=4.13.2", "uv >=0.3.2, <1.0.0"]
 
 [project.entry-points."fsspec.specs"]
-aqora = "aqora_cli.fsspec.AqoraFileSystem"
+aqora = "aqora.fsspec.AqoraFileSystem"
 
 [project.optional-dependencies]
 venv = [
@@ -35,11 +35,11 @@ Repository = "https://github.com/aqora-io/cli"
 Documentation = "https://github.com/aqora-io/cli"
 
 [project.scripts]
-aqora = "aqora_cli:main"
+aqora = "aqora:main"
 
 [tool.maturin]
 python-source = "python"
-module-name = "aqora_cli._aqora_cli"
+module-name = "aqora._aqora"
 strip = true
 features = ["extension-module"]
 

python/aqora_cli/__init__.py → python/aqora/__init__.py

@@ -1,8 +1,8 @@
 from typing_extensions import Any, TypedDict, override
 
-from . import _aqora_cli
+from . import _aqora
 from .notebook import notebook
-from aqora_cli._aqora_cli import *  # pyright: ignore[reportAssignmentType, reportWildcardImportFromLibrary]  # noqa: F403
+from aqora._aqora import *  # pyright: ignore[reportAssignmentType, reportWildcardImportFromLibrary]  # noqa: F403
 
 
 class GraphQLError(TypedDict):
@@ -29,4 +29,4 @@ def __str__(self):
         return self.message
 
 
-__doc__ = _aqora_cli.__doc__
+__doc__ = _aqora.__doc__

python/aqora_cli/fsspec.py → python/aqora/fsspec.py

@@ -7,7 +7,7 @@
 from fsspec.callbacks import DEFAULT_CALLBACK as FSSPEC_DEFAULT_CALLBACK
 from typing_extensions import Any, Unpack, TypedDict, override
 
-from ._aqora_cli import Client
+from ._aqora import Client
 
 
 @dataclass(frozen=True)

python/aqora_cli/notebook.py → python/aqora/notebook.py

@@ -8,7 +8,7 @@
 from types import ModuleType
 from typing import Any, Coroutine
 
-from ._aqora_cli import Client
+from ._aqora import Client
 
 def _parse_workspace_slug(workspace: str) -> tuple[str, str]:
     owner, slash, slug = workspace.strip().partition("/")
@@ -24,10 +24,10 @@ def _notebook_cache_dir() -> Path:
     package_dir = Path(__file__).resolve().parent
     candidates = [
         package_dir / "_workspace_notebooks",
-        Path(site.getusersitepackages()) / "aqora_cli" / "_workspace_notebooks",
+        Path(site.getusersitepackages()) / "aqora" / "_workspace_notebooks",
     ]
     for site_package in site.getsitepackages():
-        candidates.append(Path(site_package) / "aqora_cli" / "_workspace_notebooks")
+        candidates.append(Path(site_package) / "aqora" / "_workspace_notebooks")
 
     seen: set[Path] = set()
     for candidate in candidates:
@@ -51,7 +51,7 @@ def _module_name(owner: str, slug: str, filename: str) -> str:
     safe_owner = "".join(c if c.isalnum() or c == "_" else "_" for c in owner)
     safe_slug = "".join(c if c.isalnum() or c == "_" else "_" for c in slug)
     safe_file = "".join(c if c.isalnum() or c == "_" else "_" for c in Path(filename).stem)
-    return f"aqora_cli_workspace_{safe_owner}_{safe_slug}_{safe_file}"
+    return f"aqora_workspace_{safe_owner}_{safe_slug}_{safe_file}"
 
 
 def _path_part(value: str) -> str:

python/aqora_cli/pyarrow.py → python/aqora/pyarrow.py

@@ -6,7 +6,7 @@
 from pyarrow.fs import FSSpecHandler
 
 
-from aqora_cli.fsspec import AqoraFileSystem
+from aqora.fsspec import AqoraFileSystem
 
 
 def dataset(

runner/src/python.rs

@@ -291,7 +291,7 @@ impl PyEnv {
         cmd.env("VIRTUAL_ENV", path)
             .arg("pip")
             .arg("install")
-            .arg("aqora-cli[venv]");
+            .arg("aqora[venv]");
         if let Some(cache_path) = options.cache_path.as_ref() {
             cmd.arg("--cache-dir").arg(cache_path);
         }

src/commands/install.rs

@@ -250,7 +250,7 @@ pub async fn install_submission(
             let res = pip_install(
                 &env,
                 [
-                    PipPackage::pypi("aqora-cli[venv]"),
+                    PipPackage::pypi("aqora[venv]"),
                     PipPackage::tar(use_case_package_name, use_case_dir.as_os_str()),
                     PipPackage::editable(&global.project),
                 ],
@@ -302,7 +302,7 @@ pub async fn install_use_case(args: Install, global: GlobalArgs, project: PyProj
     let env = global.init_venv(&pb).await?;
 
     let mut deps = vec![
-        PipPackage::pypi("aqora-cli[venv]"),
+        PipPackage::pypi("aqora[venv]"),
         PipPackage::editable(&global.project),
     ];
 

src/python_module.rs

@@ -28,9 +28,9 @@ pub fn main(py: Python<'_>) -> PyResult<()> {
     Ok(())
 }
 
-import_exception!(aqora_cli, ClientError);
+import_exception!(aqora, ClientError);
 
-#[pyclass(frozen, name = "Client", module = "aqora_cli")]
+#[pyclass(frozen, name = "Client", module = "aqora")]
 struct PyClient {
     inner: Arc<RwLock<PyClientInner>>,
 }
@@ -232,8 +232,8 @@ impl PyClient {
 }
 
 #[pymodule]
-#[pyo3(name = "_aqora_cli")]
-pub fn aqora_cli(_: Python<'_>, m: &Bound<'_, PyModule>) -> PyResult<()> {
+#[pyo3(name = "_aqora")]
+pub fn aqora(_: Python<'_>, m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(main, m)?)?;
     m.add_class::<PipelineConfig>()?;
     m.add_class::<LayerEvaluation>()?;

template/assets/dataset_marimo/pyproject.toml.hbs

@@ -5,7 +5,7 @@ requires-python = ">={{ python_version }}"
 
 dependencies = [
   "marimo[recommended,lsp]>={{ marimo_version }}",
-  "aqora-cli[pyarrow]>={{ cli_version }}",
+  "aqora[pyarrow]>={{ cli_version }}",
 ]
 
 [dependency-groups]

template/assets/dataset_marimo/readme.py.hbs

@@ -28,7 +28,7 @@ def _():
     {{#if raw_init }}
     {{ indent raw_init '    ' }}
     {{else}}
-    from aqora_cli.pyarrow import dataset
+    from aqora.pyarrow import dataset
     data = dataset("{{ owner }}/{{ local_slug }}", "v{{ version }}")
     {{/if}}
     return (data,)

template/assets/use_case/pyproject.toml.hbs

@@ -3,7 +3,7 @@ name = "use-case"
 version = "0.1.0"
 requires-python = ">={{ python_version }}"
 
-dependencies = ["aqora-cli>={{ cli_version }}"]
+dependencies = ["aqora>={{ cli_version }}"]
 
 [build-system]
 requires = [

template/assets/use_case/use_case/use_case.py

@@ -1,6 +1,6 @@
 from types import NoneType
 from typing import Dict, AsyncIterator, List, Any
-from aqora_cli import PipelineConfig, LayerEvaluation
+from aqora import PipelineConfig, LayerEvaluation
 
 Input = NoneType
 Output = Any

test/test_aqora_client.py

@@ -2,7 +2,7 @@
 import asyncio
 import pytest
 
-from aqora_cli import Client, ClientError
+from aqora import Client, ClientError
 
 
 @pytest.mark.asyncio

test/test_aqorafs.py

@@ -1,6 +1,6 @@
 import pytest
 
-from aqora_cli.pyarrow import dataset
+from aqora.pyarrow import dataset
 
 testdata = [
     ("alice/test", "v1.0.0", (100_000, 8)),