mlflow.yaml

package:
  name: mlflow
  version: 2.17.1
  epoch: 0
  description: Open source platform for the machine learning lifecycle
  copyright:
    - license: Apache-2.0
  options:
    # Don't depend on external libraries included in virtual environments
    no-depends: true
    # Don't resolve libraries installed in virtual environments as providers
    no-provides: true
  dependencies:
    runtime:
      - bash
      - busybox
      - git
      - py3-pip
      - python3

environment:
  contents:
    packages:
      - build-base
      - busybox
      - nodejs
      - postgresql-dev
      - py3-build
      - py3-installer
      - py3-setuptools
      - py3-wheel
      - python3-dev
      - yarn

pipeline:
  - uses: git-checkout
    with:
      expected-commit: 4e711a93e8f2605710504ba40105acccc366f713
      repository: https://github.com/mlflow/mlflow
      tag: v${{package.version}}

  - runs: |
      # Build UI
      cd mlflow/server/js
      yarn install
      yarn run build
      cd -

      # Create virtual environment
      python -m venv venv --system-site-packages
      source venv/bin/activate

      # Remediate CVE-2024-1135
      sed -i "s|gunicorn<22|gunicorn<23|g" pyproject.toml

      # Install mlflow
      pip install .

      # Install dependencies
      pip install -I packaging pyparsing setuptools --no-compile

      # Remove pip
      pip uninstall --yes pip

      # Use Python in virtual environment
      sed -i "s|/home/build/venv|/usr/share/mlflow|g" venv/pyvenv.cfg
      sed -i "s|/home/build/venv|/usr/share/mlflow|g" venv/bin/*

      # Install virtual environment
      mkdir -p ${{targets.contextdir}}/usr/share/mlflow
      cp -r venv/* ${{targets.contextdir}}/usr/share/mlflow/

subpackages:
  - name: mlflow-bitnami
    description: MLflow with Bitnami additions
    options:
      no-depends: true
      no-provides: true
    dependencies:
      provides:
        - mlflow=${{package.full-version}}
      runtime:
        - bash
        - busybox
        - git
        - libpq
        - py3-pip
        - python3
    pipeline:
      - uses: bitnami/compat
        with:
          image: mlflow
          version-path: 2/debian-12
      - runs: |
          # Restore virtual env paths
          sed -i "s|/usr/share/mlflow|/home/build/venv|g" venv/pyvenv.cfg
          sed -i "s|/usr/share/mlflow|/home/build/venv|g" venv/bin/*

          # Re-initialize and activate virtual env
          python -m venv venv --system-site-packages
          source venv/bin/activate

          # Install additional dependencies used by Bitnami
          pip install -I --no-compile annotated_types anyio boto3 botocore build distlib \
            distro exceptiongroup filelock h11 httpcore httpx jmespath openai platformdirs \
            prometheus_client prometheus_flask_exporter psycopg2 pydantic pydantic_core \
            pyproject_hooks s3transfer sniffio tomli tqdm virtualenv wheel

          # Remove pip
          pip uninstall --yes pip

          # Use Python in virtual environment
          sed -i "s|/home/build/venv|/usr/share/mlflow|g" venv/pyvenv.cfg
          sed -i "s|/home/build/venv|/usr/share/mlflow|g" venv/bin/*

          # Install virtual environment
          mkdir -p ${{targets.contextdir}}/usr/share/mlflow
          cp -r venv/* ${{targets.contextdir}}/usr/share/mlflow/

          # Create auth symlink expected in Bitnami Helm chart
          mkdir -p ${{targets.contextdir}}/bitnami
          ln -sf /usr/share/mlflow/lib/python3.12/site-packages/mlflow/server/auth ${{targets.contextdir}}/bitnami/mlflow-basic-auth
    test:
      pipeline:
        - runs: |
            ls /bitnami/mlflow-basic-auth/ | grep "basic_auth.ini"
            run-script --version
            run-script --help

test:
  environment:
    contents:
      packages:
        - curl
        - git
  pipeline:
    - runs: |
        export PATH="/usr/share/mlflow/bin:${PATH}"
        python3 ./test-mlflow.py
        mlflow ui & sleep 5; curl -vsL localhost:5000

update:
  enabled: true
  ignore-regex-patterns:
    - '.*criteo.*'
    - '.*list.*'
    - '.*rc.*'
  github:
    identifier: mlflow/mlflow
    strip-prefix: v
    use-tag: true