diff options
Diffstat (limited to 'alembic/autogenerate/api.py')
| -rw-r--r-- | alembic/autogenerate/api.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/alembic/autogenerate/api.py b/alembic/autogenerate/api.py index b54e6aa..15b5b6b 100644 --- a/alembic/autogenerate/api.py +++ b/alembic/autogenerate/api.py @@ -395,7 +395,7 @@ class RevisionContext(object): # renders autogen_context.imports = set() if migration_script.imports: - autogen_context.imports.union_update(migration_script.imports) + autogen_context.imports.update(migration_script.imports) render._render_python_into_templatevars( autogen_context, migration_script, template_args ) |