Skip to content

Commit

Permalink
Update DFP MLflow ModelManager to handle model retrieval using file U…
Browse files Browse the repository at this point in the history
…RI (nv-morpheus#1578)

- Small update to handle MLflow `search_registered_models` results using file URI.

Closes nv-morpheus#1574

## 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:
  - Eli Fajardo (https://github.com/efajardo-nv)

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

URL: nv-morpheus#1578
  • Loading branch information
efajardo-nv authored Mar 26, 2024
1 parent 83025e1 commit 2fb97ae
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -225,7 +225,7 @@ def _model_exists(self, reg_model_name: str, timeout: float = 1.0) -> bool:

self._existing_models.update(model.name for model in results)

if (len(results.token) == 0):
if (results.token is None or len(results.token) == 0):
break

logger.debug("Updating list of available models... Done.")
Expand Down

0 comments on commit 2fb97ae

Please sign in to comment.