<p dir="auto"><strong>Apache Airflow version</strong>:  2.0.1</p>
<p dir="auto"><strong>Environment</strong>:</p>
<ul dir="auto">
<li><strong>Cloud provider or hardware configuration</strong>:  Google Cloud Platform Compute Engine</li>
<li><strong>OS</strong> (e.g. from /etc/os-release): NAME="Ubuntu" VERSION="20.04.2 LTS (Focal Fossa)"</li>
<li><strong>Kernel</strong> (e.g. <code class="notranslate">uname -a</code>): Linux finance-airflow 5.4.0-1041-gcp <a class="issue-link js-issue-link" data-error-text="Failed to load title" data-id="89200849" data-permission-text="Title is private" data-url="https://github.com/apache/airflow/issues/44" data-hovercard-type="pull_request" data-hovercard-url="/apache/airflow/pull/44/hovercard" href="https://github.com/apache/airflow/pull/44">#44</a>-Ubuntu SMP Fri Mar 26 03:53:35 UTC 2021 x86_64 x86_64 x86_64 GNU/Linux</li>
<li><strong>Install tools</strong>: pip</li>
</ul>
<p dir="auto"><strong>What happened</strong>:</p>
<p dir="auto">Installed with usual <code class="notranslate">pip install apache-airflow</code> which installed <code class="notranslate">sqlalchemy 1.4.6</code>, since <code class="notranslate">apache-airflow 2.0.1</code> requires <code class="notranslate">sqlalchemy&lt;2,&gt;=1.3.18</code></p>
<p dir="auto">Ran:</p>
<p dir="auto"><code class="notranslate">airflow db init</code></p>
<p dir="auto"><code class="notranslate">from sqlalchemy.orm.query import _ColumnEntity ImportError: cannot import name '_ColumnEntity' from 'sqlalchemy.orm.query' (/home/foo/airflow/.direnv/python-3.8.5/lib/python3.8/site-packages/sqlalchemy/orm/query.py)</code></p>
<p dir="auto"><strong>What you expected to happen</strong>:</p>
<p dir="auto">Initializing <code class="notranslate">sqlite3</code> database for testing</p>
<p dir="auto"><strong>How to reproduce it</strong>:</p>
<p dir="auto"><code class="notranslate">pip install apache-airflow</code> on Ubuntu 20.04.2 LTS</p>
<p dir="auto"><strong>Anything else we need to know</strong>:</p>
<p dir="auto">Downgraded <code class="notranslate">sqlalchemy</code> via <code class="notranslate">pip install --upgrade sqlalchemy==1.3.18</code>, which lets database initialize, but one expects dependencies to be handled by <code class="notranslate">setup.py</code>. Suggest updating <code class="notranslate">setup.py</code> for <code class="notranslate">apache-airflow</code> version in <code class="notranslate">pypi</code>.</p>
<p dir="auto"><strong>Apache Airflow version</strong>: 2.0.1</p>
<p dir="auto"><strong>Kubernetes version (if you are using kubernetes)</strong> (use <code class="notranslate">kubectl version</code>): N/A</p>
<p dir="auto"><strong>Environment</strong>:</p>
<ul dir="auto">
<li><strong>Cloud provider or hardware configuration</strong>:</li>
<li><strong>OS</strong> (e.g. from /etc/os-release): Mac OS Big Sur</li>
<li><strong>Kernel</strong> (e.g. <code class="notranslate">uname -a</code>):</li>
<li><strong>Install tools</strong>: pip 20.1.1</li>
<li><strong>Others</strong>:</li>
</ul>
<p dir="auto"><strong>What happened</strong>:</p>
<p dir="auto">Our CI environment broke due to the release of SQLAlchemy 1.4, which is incompatible with the latest version of sqlalchemy-utils. (<a href="https://github.com/kvesteri/sqlalchemy-utils/issues/505" data-hovercard-type="issue" data-hovercard-url="/kvesteri/sqlalchemy-utils/issues/505/hovercard">Related issue</a>)</p>
<p dir="auto">Partial stacktrace:</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="  File &quot;/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/airflow/www/utils.py&quot;, line 27, in &lt;module&gt;
    from flask_appbuilder.models.sqla.interface import SQLAInterface
  File &quot;/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/flask_appbuilder/models/sqla/interface.py&quot;, line 16, in &lt;module&gt;
    from sqlalchemy_utils.types.uuid import UUIDType
  File &quot;/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy_utils/__init__.py&quot;, line 1, in &lt;module&gt;
    from .aggregates import aggregated  # noqa
  File &quot;/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy_utils/aggregates.py&quot;, line 372, in &lt;module&gt;
    from .functions.orm import get_column_key
  File &quot;/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy_utils/functions/__init__.py&quot;, line 1, in &lt;module&gt;
    from .database import (  # noqa
  File &quot;/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy_utils/functions/database.py&quot;, line 11, in &lt;module&gt;
    from .orm import quote
  File &quot;/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy_utils/functions/orm.py&quot;, line 14, in &lt;module&gt;
    from sqlalchemy.orm.query import _ColumnEntity
ImportError: cannot import name '_ColumnEntity' from 'sqlalchemy.orm.query' (/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy/orm/query.py)"><pre class="notranslate"><code class="notranslate">  File "/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/airflow/www/utils.py", line 27, in &lt;module&gt;
    from flask_appbuilder.models.sqla.interface import SQLAInterface
  File "/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/flask_appbuilder/models/sqla/interface.py", line 16, in &lt;module&gt;
    from sqlalchemy_utils.types.uuid import UUIDType
  File "/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy_utils/__init__.py", line 1, in &lt;module&gt;
    from .aggregates import aggregated  # noqa
  File "/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy_utils/aggregates.py", line 372, in &lt;module&gt;
    from .functions.orm import get_column_key
  File "/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy_utils/functions/__init__.py", line 1, in &lt;module&gt;
    from .database import (  # noqa
  File "/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy_utils/functions/database.py", line 11, in &lt;module&gt;
    from .orm import quote
  File "/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy_utils/functions/orm.py", line 14, in &lt;module&gt;
    from sqlalchemy.orm.query import _ColumnEntity
ImportError: cannot import name '_ColumnEntity' from 'sqlalchemy.orm.query' (/Users/samwheating/Desktop/tmp_venv/lib/python3.7/site-packages/sqlalchemy/orm/query.py)
</code></pre></div>
<p dir="auto">I'm not sure what the typical procedure is in the case of breaking changes to dependencies, but seeing as there's an upcoming release I thought it might be worth pinning sqlalchemy to 1.3.x? (Or pin the version of sqlalchemy-utils to a compatible version if one is released before Airflow 2.0.2)</p>
<p dir="auto"><strong>What you expected to happen</strong>:</p>
<p dir="auto"><code class="notranslate">airflow db init</code> to run successfully.</p>

<p dir="auto"><strong>How to reproduce it</strong>:</p>
<ol dir="auto">
<li>Create a new virtualenv</li>
<li><code class="notranslate">pip install apache-airflow</code></li>
<li><code class="notranslate">airflow db init</code></li>
</ol>