From 34780bfa4c861b22f2e1830152ae741e046e44d8 Mon Sep 17 00:00:00 2001 From: nnshah1 Date: Tue, 16 Apr 2024 09:28:45 -0700 Subject: [PATCH 1/2] updated to add models loaded dynamically but not in a physical repository --- .../model_repository_manager.cc | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/src/model_repository_manager/model_repository_manager.cc b/src/model_repository_manager/model_repository_manager.cc index 8b06ad914..7148c090f 100644 --- a/src/model_repository_manager/model_repository_manager.cc +++ b/src/model_repository_manager/model_repository_manager.cc @@ -1028,8 +1028,8 @@ Status ModelRepositoryManager::RepositoryIndex( const bool ready_only, std::vector* index) { - std::set seen_models; - std::set duplicate_models; + std::unordered_set seen_models; + std::unordered_set duplicate_models; for (const auto& repository_path : repository_paths_) { const std::string model_namespace = (enable_model_namespacing_ ? repository_path : ""); @@ -1059,6 +1059,18 @@ ModelRepositoryManager::RepositoryIndex( } } + // Any loaded models which are + // not present in the local model repository + // are added to the index directly + + for (const auto& mapping_it : global_map_) { + for (const auto& model_id : mapping_it.second) { + if (seen_models.count(model_id) == 0) { + seen_models.insert(model_id); + } + } + } + ModelStateMap states = ModelStates(); for (const auto& model_id : seen_models) { From 6a2404e11060b268caf60be788f32e017d297ac0 Mon Sep 17 00:00:00 2001 From: nnshah1 Date: Tue, 16 Apr 2024 10:18:50 -0700 Subject: [PATCH 2/2] updated with pre commit hook --- src/model_repository_manager/model_repository_manager.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/model_repository_manager/model_repository_manager.cc b/src/model_repository_manager/model_repository_manager.cc index 7148c090f..02d115c79 100644 --- a/src/model_repository_manager/model_repository_manager.cc +++ b/src/model_repository_manager/model_repository_manager.cc @@ -1062,11 +1062,11 @@ ModelRepositoryManager::RepositoryIndex( // Any loaded models which are // not present in the local model repository // are added to the index directly - + for (const auto& mapping_it : global_map_) { for (const auto& model_id : mapping_it.second) { if (seen_models.count(model_id) == 0) { - seen_models.insert(model_id); + seen_models.insert(model_id); } } }