diff --git a/extension/named_data_map/merged_data_map.cpp b/extension/named_data_map/merged_data_map.cpp index 63ef884892b..630395e006c 100644 --- a/extension/named_data_map/merged_data_map.cpp +++ b/extension/named_data_map/merged_data_map.cpp @@ -69,11 +69,10 @@ ET_NODISCARD Result MergedDataMap::get_tensor_layout( ET_NODISCARD Result MergedDataMap::get_data(string_view key) const { const auto it = key_to_map_index_.find(key.data()); - ET_CHECK_OR_RETURN_ERROR( - it != key_to_map_index_.end(), - NotFound, - "Key %s not found in named data maps", - key.data()); + if (it == key_to_map_index_.end()) { + ET_LOG(Debug, "Key %s not found in named data maps.", key.data()); + return Error::NotFound; + } return named_data_maps_.at(it->second)->get_data(key); }