diff --git a/pyproject.toml b/pyproject.toml index c1850932..051b57ff 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -148,7 +148,7 @@ shell = "find . -name '*.ipynb' | awk 'system(\"git check-ignore --quiet \" $0)' [tool.poe.tasks.lint] help = "Runs all formatting, lints, and checks (fixing where possible)" -sequence = [ "docs-build", "lock-fix", "fmt-fix", "lint-fix", "type-check", "dep-check" ] +sequence = [ "lock-fix", "fmt-fix", "lint-fix", "type-check", "dep-check", "docs-build" ] [tool.poe.tasks.test-gr] help = "Runs graph-retriever tests (against in-memory stores)" diff --git a/scripts/drop-astra-keyspace.py b/scripts/drop-astra-keyspace.py index 38844e86..777f8d69 100644 --- a/scripts/drop-astra-keyspace.py +++ b/scripts/drop-astra-keyspace.py @@ -1,10 +1,15 @@ import os +import sys from astrapy import AstraDBDatabaseAdmin from astrapy.authentication import StaticTokenProvider token = StaticTokenProvider(os.environ["ASTRA_DB_APPLICATION_TOKEN"]) -keyspace = os.environ.get("ASTRA_DB_KEYSPACE", "default_keyspace") +keyspace = os.environ.get("ASTRA_DB_KEYSPACE") +if keyspace is None: + print("No keyspace to drop.") # noqa: T201 + sys.exit() + api_endpoint = os.environ["ASTRA_DB_API_ENDPOINT"] admin = AstraDBDatabaseAdmin(api_endpoint=api_endpoint, token=token) diff --git a/uv.lock b/uv.lock index 57bdeb51..cc1f69e6 100644 --- a/uv.lock +++ b/uv.lock @@ -1315,8 +1315,8 @@ dev = [ [package.metadata] requires-dist = [ { name = "numpy", specifier = ">=1.26.4" }, + { name = "pytest", specifier = ">=8.3.4" }, { name = "pytest", marker = "extra == 'testing'", specifier = ">=8.3.4" }, - { name = "pytest", extras = ["testing"], specifier = ">=8.3.4" }, { name = "simsimd", marker = "extra == 'simsimd'", specifier = ">=6.2.1" }, { name = "typing-extensions", specifier = ">=4.12.2" }, ]