Update src/sqlacodegen/generators.py

Co-authored-by: Alex Grönholm <alex.gronholm@nextday.fi>
agronholm · Jan 11, 2025 · 79c25af · 79c25af
1 parent ecc1b89
commit 79c25af
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/sqlacodegen/generators.py b/src/sqlacodegen/generators.py
@@ -1385,6 +1385,7 @@ def collect_imports(self, models: Iterable[Model]) -> None:
         super(DeclarativeGenerator, self).collect_imports(models)
         if any(isinstance(model, Model) for model in models):
             self.add_literal_import("sqlalchemy", "MetaData")
+
         if any(isinstance(model, ModelClass) for model in models):
             self.add_literal_import("sqlmodel", "SQLModel")
             self.add_literal_import("sqlmodel", "Field")