Mlflow Plugin Fix (#6685) (#6704)

* Mlflow plugin fix Co-authored-by: Francesco Petrini <francescogpetrini@gmail.com>
triton-inference-server · Dec 14, 2023 · bbd0d1a · bbd0d1a
1 parent ea6d71e
commit bbd0d1a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deploy/mlflow-triton-plugin/mlflow_triton/deployments.py b/deploy/mlflow-triton-plugin/mlflow_triton/deployments.py
@@ -338,8 +338,9 @@ def _get_copy_paths(self, artifact_path, name, flavor):
             # with proper model versions and version strategy, which may differ from
             # the versioning in MLFlow
             for file in artifact_path.iterdir():
-                if file.name not in ["MLmodel", "conda.yaml"]:
+                if file.is_dir():
                     copy_paths["model_path"]["from"] = file
+                    break
             copy_paths["model_path"]["to"] = triton_deployment_dir
         elif flavor == "onnx":
             # Look for model file via MLModel metadata or iterating dir