Merge pull request #40 from neptune-ai/aw/update-integration

Update MLflow integration

Author: AleksanderWWW

.github/actions/e2e/action.yml

@@ -0,0 +1,26 @@
+---
+name: Test E2E
+description: Run e2e tests
+inputs:
+  working_directory:
+    description: "Working directory"
+    required: false
+    default: .
+runs:
+  using: "composite"
+  steps:
+    - name: Checkout
+      uses: actions/checkout@v2
+      with:
+        repository: neptune-ai/neptune-mlflow
+        path: ${{ inputs.working_directory }}
+
+    - name: Install dependencies
+      working-directory: ${{ inputs.working_directory }}
+      run: pip install -e .[dev]
+      shell: bash
+
+    - name: Run tests
+      working-directory: ${{ inputs.working_directory }}
+      run: pytest -v tests/e2e
+      shell: bash

.github/actions/unit/action.yml

@@ -0,0 +1,26 @@
+---
+name: Unit test
+description: Run unit tests
+inputs:
+  working_directory:
+    description: "Working directory"
+    required: false
+    default: .
+runs:
+  using: "composite"
+  steps:
+    - name: Checkout
+      uses: actions/checkout@v2
+      with:
+        repository: neptune-ai/neptune-mlflow
+        path: ${{ inputs.working_directory }}
+
+    - name: Install dependencies
+      working-directory: ${{ inputs.working_directory }}
+      run: pip install -e .[dev]
+      shell: bash
+
+    - name: Run tests
+      working-directory: ${{ inputs.working_directory }}
+      run: pytest -v tests/unit
+      shell: bash

.github/workflows/ci.yml

@@ -0,0 +1,79 @@
+name: neptune-mlflow
+on: [push]
+jobs:
+  pre-commit:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v2
+
+      - uses: actions/setup-python@v2
+        with:
+          python-version: 3.9
+
+      - name: Install dependencies
+        run: |
+          pip install -e .[dev]
+
+      - name: Pre-commit
+        run: |
+          pre-commit run --show-diff-on-failure --color=always
+
+  e2e:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-latest, macos-latest, windows-latest]
+        python-version: [3.9]
+    steps:
+      - uses: actions/checkout@v2
+
+      - uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Test
+        env:
+          NEPTUNE_API_TOKEN: ${{ secrets.E2E_NEPTUNE_API_TOKEN }}
+          NEPTUNE_PROJECT: e2e-tests/e2e
+        uses: ./.github/actions/e2e
+
+  unit:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ ubuntu-latest, macos-latest, windows-latest ]
+        python-version: [ 3.9 ]
+    steps:
+      - uses: actions/checkout@v2
+
+      - uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Test
+        uses: ./.github/actions/unit
+
+  publish:
+    needs: [pre-commit, e2e, unit]
+    runs-on: ubuntu-latest
+    if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags')
+    steps:
+      - uses: actions/checkout@v2
+
+      - uses: actions/setup-python@v2
+        with:
+          python-version: 3.9
+
+      - name: Install build dependencies
+        run: pip install poetry poetry-dynamic-versioning
+
+      - name: Build package
+        run : |
+          poetry build
+
+      - name: Publish package
+        uses: pypa/gh-action-pypi-publish@v1.5.1
+        with:
+          user: __token__
+          password: ${{ secrets.PYPI_API_TOKEN }}
+          packages_dir: dist

.github/workflows/pre-commit.yml

@@ -107,3 +107,10 @@ venv.bak/
 
 # Pycharm
 .idea/
+
+# MLflow
+mlruns/
+test_tracking_uri/
+
+# Neptune
+.neptune/

.gitignore

@@ -0,0 +1,4 @@
+## [UNRELEASED] neptune-mlflow 1.0.0
+
+### Changes
+- Updated MLflow exporter to work with the current neptune.ai API ([#40](https://github.com/neptune-ai/neptune-mlflow/pull/40))