Skip to content

Commit 8bf1f71

Browse files
authored
feat(ingestion/mlflow): improve mlflow connector to pull run and experiments (#12587)
1 parent 598cda2 commit 8bf1f71

File tree

9 files changed

+1073
-53
lines changed

9 files changed

+1073
-53
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
source:
2+
type: mlflow
3+
config:
4+
tracking_uri: "http://127.0.0.1:5000"
5+
6+
sink:
7+
type: datahub-rest
8+
config:
9+
server: "http://localhost:8080"

metadata-ingestion/src/datahub/ingestion/source/common/subtypes.py

+5
Original file line numberDiff line numberDiff line change
@@ -92,3 +92,8 @@ class BIAssetSubTypes(StrEnum):
9292
# SAP Analytics Cloud
9393
SAC_STORY = "Story"
9494
SAC_APPLICATION = "Application"
95+
96+
97+
class MLAssetSubTypes(StrEnum):
98+
MLFLOW_TRAINING_RUN = "ML Training Run"
99+
MLFLOW_EXPERIMENT = "ML Experiment"

0 commit comments

Comments
 (0)