error when using batch_op.create_foreign_key()

[harishsurf]

Hi, I am using batch_op.create_foreign_key() in my alembic migration script:

with op.batch_alter_table("table1") as batch_op:
    batch_op.create_foreign_key(
          op.f("fk_table1_col1_table2"), 
          "table2",
          ["col1"],
          ["col2"],
    )

as mentioned in the docs but I get the following error:
sqlalchemy.exc.ArgumentError: contains_column cannot be used with string arguments. Use ``col_name in table.c`` instead.
why does it error and what would the argument look like?

[zzzeek]

can you provide a complete stack trace?

[harishsurf]

  File "/sample_proj/data/migrations/versions/61cadbve9fc_tiggers_.py", line 55, in upgrade
    batch_op.create_foreign_key(
  File "/usr/local/Cellar/python@3.9/3.9.17/Frameworks/Python.framework/Versions/3.9/lib/python3.9/contextlib.py", line 126, in __exit__
    next(self.gen)
  File "/usr/local/lib/python3.9/site-packages/alembic/operations/base.py", line 325, in batch_alter_table
    impl.flush()
  File "/usr/local/lib/python3.9/site-packages/alembic/operations/batch.py", line 108, in flush
    batch_impl._create(self.impl)
  File "/usr/local/lib/python3.9/site-packages/alembic/operations/batch.py", line 284, in _create
    self._transfer_elements_to_new_table()
  File "/usr/local/lib/python3.9/site-packages/alembic/operations/batch.py", line 235, in _transfer_elements_to_new_table
    self._setup_referent(m, const)
  File "/usr/local/lib/python3.9/site-packages/alembic/operations/batch.py", line 267, in _setup_referent
    if not t.c.contains_column(colname):
  File "/usr/local/lib/python3.9/site-packages/sqlalchemy/sql/base.py", line 1337, in contains_column
    raise exc.ArgumentError(
sqlalchemy.exc.ArgumentError: contains_column cannot be used with string arguments. Use ``col_name in table.c`` instead.

[zzzeek]

you're on an old version of alembic that is not up to date for the version of SQLAlchemy you are running, so I would ensure you are on the latest alembic version that supports the SQLAlchemy version you have in use.

[harishsurf]

thanks, that was it!