fix passing kwargs to alembic

dymmond · Oct 21, 2024 · 4f20e09 · 4f20e09
1 parent 80c297b
commit 4f20e09
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/edgy/cli/base.py b/edgy/cli/base.py
@@ -67,7 +67,7 @@ def __init__(
  render_as_batch: bool = True,
  **kwargs: Any,
  ) -> None:
- super().__init__(**kwargs)
+ super().__init__()
 
  self.app = app
  self.configure_callbacks: list[Callable] = []
@@ -95,9 +95,11 @@ def __init__(
  self.registry.refresh_metadata()
 
  self.directory = "migrations"
- self.alembic_ctx_kwargs = kwargs
- self.alembic_ctx_kwargs["compare_type"] = compare_type
- self.alembic_ctx_kwargs["render_as_batch"] = render_as_batch
+ self.alembic_ctx_kwargs = {
+ **kwargs,
+ "compare_type": compare_type,
+ "render_as_batch": render_as_batch,
+ }
 
  self.set_edgy_extension(app)