Importer will now use the most recently created artifact when reusing…

… an existing artifact instead of the one with the highest ID. PiperOrigin-RevId: 479241179
rickqiu · Oct 6, 2022 · 0d68333 · 0d68333
1 parent 14c1518
commit 0d68333
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/RELEASE.md b/RELEASE.md
@@ -17,6 +17,7 @@
     `[tflite-support]` extra.
 *   Statistics gen and Schema gen now crash on empty input examples and statistics respectively.
 *   Importer will now check that an existing artifact has the same type as the intended output before reusing the existing artifact.
+*   Importer will now use the most recently created artifact when reusing an existing artifact instead of the one with the highest ID.
 
 ## Dependency Updates
 

diff --git a/tfx/dsl/components/common/importer.py b/tfx/dsl/components/common/importer.py
@@ -143,7 +143,8 @@ def _prepare_artifact(
     # not explicitly ask for reimport, reuse that artifact.
     if previous_artifacts:
       absl.logging.info('Reusing existing artifact')
-      result.set_mlmd_artifact(max(previous_artifacts, key=lambda m: m.id))
+      result.set_mlmd_artifact(
+          max(previous_artifacts, key=lambda m: m.create_time_since_epoch))
 
   return result