Skip to content

Commit

Permalink
Fix tests to support mlflow v2.7 (nv-morpheus#1220)
Browse files Browse the repository at this point in the history
Closes nv-morpheus#1199

## By Submitting this PR I confirm:
- I am familiar with the [Contributing Guidelines](https://github.com/nv-morpheus/Morpheus/blob/main/docs/source/developer_guide/contributing.md).
- When the PR is ready for review, new or existing tests cover these changes.
- When the PR is ready for review, the documentation is up to date with these changes.

Authors:
  - David Gardner (https://github.com/dagardner-nv)

Approvers:
  - Michael Demoret (https://github.com/mdemoret-nv)

URL: nv-morpheus#1220
  • Loading branch information
dagardner-nv authored Sep 26, 2023
1 parent c8f4e16 commit 7655b93
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 1 deletion.
2 changes: 1 addition & 1 deletion docker/conda/environments/cuda11.8_dev.yml
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ dependencies:
- jupyterlab
- libgrpc>=1.49
- librdkafka=1.9.2
- mlflow>=2.2.1,<2.7
- mlflow>=2.2.1,<3
- mrc=23.11
- networkx=3.1
- ninja=1.10
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,8 @@
from collections import namedtuple
from unittest import mock

# This import is needed to ensure our mocks work correctly, otherwise mlflow.pytorch is a stub.
import mlflow.pytorch # noqa: F401 pylint: disable=unused-import
import pandas as pd
import pytest

Expand Down

0 comments on commit 7655b93

Please sign in to comment.