fix: fix mlflow test cases (#98)

Signed-off-by: s0nicboOm <[email protected]>

.flake8

@@ -1,6 +1,5 @@
 [flake8]
-ignore = |
- E203, F821
+ignore = E203, F821
 exclude = .git,__pycache__,docs/source/conf.py,old,build,dist
 max-complexity = 10
 max-line-length = 100

examples/numalogic-simple-pipeline/Dockerfile

@@ -1,8 +1,9 @@
-FROM python:3.9.12-slim
+FROM python:3.10-slim-bullseye AS builder
+
 ENV PYTHONFAULTHANDLER=1 \
  PYTHONUNBUFFERED=1 \
  PYTHONHASHSEED=random \
- PIP_NO_CACHE_DIR=off \
+ PIP_NO_CACHE_DIR=on \
  PIP_DISABLE_PIP_VERSION_CHECK=on \
  PIP_DEFAULT_TIMEOUT=100 \
  POETRY_VERSION=1.2.2 \
@@ -13,28 +14,49 @@ ENV PYTHONFAULTHANDLER=1 \
  VENV_PATH="/opt/pysetup/.venv"
 
 ENV PATH="$POETRY_HOME/bin:$VENV_PATH/bin:$PATH"
+
 RUN apt-get update \
  && apt-get install --no-install-recommends -y \
  curl \
  wget \
  # deps for building python deps
- build-essential
-RUN apt-get install -y git
-# install poetry - respects $POETRY_VERSION & $POETRY_HOME
-RUN curl -sSL https://install.python-poetry.org | python3 -
+ build-essential \
+ && apt-get install -y git \
+ && apt-get clean && rm -rf /var/lib/apt/lists/* \
+ \
+ # install dumb-init
+ && wget -O /dumb-init https://github.com/Yelp/dumb-init/releases/download/v1.2.5/dumb-init_1.2.5_x86_64 \
+ && chmod +x /dumb-init \
+ && curl -sSL https://install.python-poetry.org | python3 -
+
+FROM builder AS mlflow
 
 WORKDIR $PYSETUP_PATH
-ADD . $PYSETUP_PATH
-WORKDIR $PYSETUP_PATH
-RUN poetry install --all-extras
+COPY ./pyproject.toml ./poetry.lock ./
+RUN poetry install --only mlflow --no-cache --no-root && \
+ rm -rf ~/.cache/pypoetry/
 
 ADD . /app
+WORKDIR /app
+
+RUN chmod +x entry.sh
+
+ENTRYPOINT ["/dumb-init", "--"]
+CMD ["/app/entry.sh"]
+
+EXPOSE 5000
+
 
-# install dumb-init
-RUN wget -O /dumb-init https://github.com/Yelp/dumb-init/releases/download/v1.2.5/dumb-init_1.2.5_x86_64
-RUN chmod +x /dumb-init
+FROM builder AS udf
 
+WORKDIR $PYSETUP_PATH
+COPY ./pyproject.toml ./poetry.lock ./
+RUN poetry install --with mlflow-skinny --no-cache --no-root && \
+ rm -rf ~/.cache/pypoetry/
+
+ADD . /app
 WORKDIR /app
+
 RUN chmod +x entry.sh
 
 ENTRYPOINT ["/dumb-init", "--"]