Skip to content

Commit e9a58c8

Browse files
committed
fix(airflow): dag and docker with Variable and apache-airflow
1 parent 31aaab9 commit e9a58c8

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

src/airflow/src/airflow_inference.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22
import logging
33
import os
44

5+
from airflow.models import Variable
6+
57
BASE_URL = os.getenv('INFERENCE_URL')
68
NEO4J_DB = os.getenv('NEO4J_DATABASE')
79
API_TOKEN = os.getenv('API_TOKEN')

tools/images/Dockerfile-airflow

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,9 @@ FROM python:3.9-slim
22

33
WORKDIR /app
44

5-
RUN pip install --no-cache-dir requests
5+
RUN pip install --no-cache-dir "apache-airflow==2.9.1" \
6+
--constraint "https://raw.githubusercontent.com/apache/airflow/constraints-2.9.1/constraints-3.9.txt" \
7+
requests
68

79
COPY ../../src/airflow/src/airflow_inference.py .
810

0 commit comments

Comments
 (0)