ashb commented on a change in pull request #15330:
URL: https://github.com/apache/airflow/pull/15330#discussion_r708162445
##########
File path: airflow/decorators/__init__.py
##########
@@ -15,34 +15,58 @@
# specific language governing permissions and limitations
# under the License.
-from typing import Callable, Optional
-
-from airflow.decorators.python import python_task
-from airflow.decorators.python_virtualenv import _virtualenv_task
+from airflow.decorators.python import PythonDecoratorMixin
+from airflow.decorators.python_virtualenv import PythonVirtualenvDecoratorMixin
from airflow.decorators.task_group import task_group # noqa
from airflow.exceptions import AirflowException
from airflow.models.dag import dag # noqa
from airflow.providers_manager import ProvidersManager
+# [START import_docker]
+try:
+ from airflow.providers.docker.decorators.docker import DockerDecoratorMixin
+except ImportError:
+ DockerDecoratorMixin = None
+# [END import_docker]
+
class _TaskDecorator:
Review comment:
Typo, I meant to say
```python
class _TaskDecorator(PythonDecoratorMixin, PythonVirtualenvDecoratorMixin)
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
