diff --git a/ml_metadata/metadata_store/postgresql_query_executor.cc b/ml_metadata/metadata_store/postgresql_query_executor.cc index ab0db89d4..3606b6873 100644 --- 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 index 094a7abd4..fec5c1c0f 100644 --- 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());