add missing columns to query executor sel stmt

Signed-off-by: Isabella Basso do Amaral <[email protected]>
opendatahub-io · Oct 2, 2024 · 31e44df · 31e44df
1 parent b951b0a
commit 31e44df
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ml_metadata/metadata_store/postgresql_query_executor.cc b/ml_metadata/metadata_store/postgresql_query_executor.cc
@@ -903,7 +903,7 @@ absl::Status PostgreSQLQueryExecutor::ListNodeIDsUsingOptions(
                        sql_gen_status.message()));
     }
     sql_query = absl::Substitute(
-        "SELECT distinct $0.id, $0.create_time_since_epoch FROM $1 WHERE $2 "
+        "SELECT distinct $0.id, $0.create_time_since_epoch, $0.last_update_time_since_epoch FROM $1 WHERE $2 "
         "AND ",
         *node_table_alias,
         // TODO(b/257334039): remove query_version-conditional logic

diff --git a/ml_metadata/metadata_store/query_config_executor.cc b/ml_metadata/metadata_store/query_config_executor.cc
@@ -844,7 +844,7 @@ absl::Status QueryConfigExecutor::ListNodeIDsUsingOptions(
                        sql_gen_status.message()));
     }
     sql_query = absl::Substitute(
-        "SELECT distinct $0.`id` FROM $1 WHERE $2 AND ", *node_table_alias,
+        "SELECT distinct $0.`id`, $0.`create_time_since_epoch`, $0.`last_update_time_since_epoch` FROM $1 WHERE $2 AND ", *node_table_alias,
         // TODO(b/257334039): remove query_version-conditional logic
         query_builder.GetFromClause(query_version),
         query_builder.GetWhereClause());