fix(airflow): variable import from script to dag

Author: vancauwe

src/airflow/dags/dag-inference.py

@@ -2,6 +2,8 @@
 from airflow.providers.docker.operators.docker import DockerOperator
 from datetime import datetime, timedelta
 
+from airflow.models import Variable
+
 default_args = {
     'owner': 'airflow',
     'retries': 1,

src/airflow/src/airflow_inference.py

@@ -2,8 +2,6 @@
 import logging
 import os
 
-from airflow.models import Variable
-
 BASE_URL = os.getenv('INFERENCE_URL')
 NEO4J_DB = os.getenv('NEO4J_DATABASE')
 API_TOKEN = os.getenv('API_TOKEN')

tools/images/Dockerfile-airflow

@@ -2,9 +2,7 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-RUN pip install --no-cache-dir "apache-airflow==2.9.1" \
-    --constraint "https://raw.githubusercontent.com/apache/airflow/constraints-2.9.1/constraints-3.9.txt" \
-    requests
+RUN pip install --no-cache-dir requests
 
 COPY ../../src/airflow/src/airflow_inference.py .