diff options
Diffstat (limited to 'alembic/operations/base.py')
| -rw-r--r-- | alembic/operations/base.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/alembic/operations/base.py b/alembic/operations/base.py index 59bbfc4..07ddd5a 100644 --- a/alembic/operations/base.py +++ b/alembic/operations/base.py @@ -17,9 +17,11 @@ from . import batch from . import schemaobj from .. import util from ..util import sqla_compat +from ..util.compat import formatannotation_fwdref from ..util.compat import inspect_formatargspec from ..util.compat import inspect_getfullargspec + NoneType = type(None) if TYPE_CHECKING: @@ -121,7 +123,9 @@ class Operations(util.ModuleClsProxy): name_args[0:2] = ["self"] - args = inspect_formatargspec(*spec) + args = inspect_formatargspec( + *spec, formatannotation=formatannotation_fwdref + ) num_defaults = len(spec[3]) if spec[3] else 0 if num_defaults: defaulted_vals = name_args[0 - num_defaults :] @@ -134,6 +138,7 @@ class Operations(util.ModuleClsProxy): spec[2], defaulted_vals, formatvalue=lambda x: "=" + x, + formatannotation=formatannotation_fwdref, ) args = re.sub( |