[MLMD] Backport MLMD Utils and APIs added to Pipelines since Jan 8 (#23)

* Backport changes to ArtifactCustomProperties and ExecutionCustomProperties * Backport Util and API changes needed to merge LineageView into KFP
kubeflow · Jan 28, 2020 · 756476a · 756476a
1 parent 963164a
commit 756476a
Show file tree

Hide file tree

Showing 12 changed files with 121 additions and 14 deletions.
diff --git a/build/lib/index.js b/build/lib/index.js
diff --git a/build/lib/index.js.map b/build/lib/index.js.map
diff --git a/build/lib/mlmd/Api.js b/build/lib/mlmd/Api.js
diff --git a/build/lib/mlmd/Api.js.map b/build/lib/mlmd/Api.js.map
diff --git a/build/lib/mlmd/LineageApi.js b/build/lib/mlmd/LineageApi.js
diff --git a/build/lib/mlmd/LineageApi.js.map b/build/lib/mlmd/LineageApi.js.map
diff --git a/build/lib/mlmd/Utils.js b/build/lib/mlmd/Utils.js
diff --git a/build/lib/mlmd/Utils.js.map b/build/lib/mlmd/Utils.js.map
diff --git a/src/index.ts b/src/index.ts
@@ -43,7 +43,7 @@ export {
   ExecutionProperties,
   ExecutionCustomProperties,
 } from './mlmd/Api'
-export {getArtifactTypes} from './mlmd/LineageApi'
+export {getArtifactTypes, getArtifactCreationTime} from './mlmd/LineageApi'
 export {LineageView} from './mlmd/LineageView'
 export {getMetadataValue, getResourceProperty, getTypeName} from './mlmd/Utils'
 export {logger, formatDateString, titleCase} from './common/Utils'
diff --git a/src/mlmd/Api.ts b/src/mlmd/Api.ts
@@ -14,6 +14,9 @@ export enum ArtifactProperties {
 export enum ArtifactCustomProperties {
   WORKSPACE = '__kf_workspace__',
   RUN = '__kf_run__',
+  NAME = 'name',
+  PIPELINE_NAME = 'pipeline_name', // TODO: Remove when switching to contexts
+  RUN_ID = 'run_id', // TODO: Remove when switching to contexts
 }
 
 /** Known Execution properties */
@@ -27,6 +30,8 @@ export enum ExecutionProperties {
 /** Known Execution custom properties */
 export enum ExecutionCustomProperties {
   WORKSPACE = '__kf_workspace__',
+  RUN_ID = 'run_id', // TODO: Remove when switching to contexts
+  TASK_ID = 'task_id',
 }
 
 /** Format for a List request */