File tree Expand file tree Collapse file tree 2 files changed +10
-2
lines changed Expand file tree Collapse file tree 2 files changed +10
-2
lines changed Original file line number Diff line number Diff line change 55from util import daemon
66from . import util
77from . import connector
8+ from controller .misc import manager as misc
89from controller .model_provider import manager as model_manager
910
1011
1112def get_recommended_encoders () -> List [Any ]:
1213 recommendations = connector .request_listing_recommended_encoders ()
13- existing_models = model_manager .get_model_provider_info ()
14+ if misc .check_is_managed :
15+ existing_models = model_manager .get_model_provider_info ()
16+ else :
17+ existing_models = []
1418 for model in existing_models :
1519
1620 if not model ["zero_shot_pipeline" ]:
Original file line number Diff line number Diff line change 2323from util import daemon
2424from controller .weak_supervision import weak_supervision_service as weak_supervision
2525from controller .model_provider import manager as model_manager
26+ from controller .misc import manager as misc
2627
2728
2829def get_zero_shot_text (
@@ -47,7 +48,10 @@ def get_zero_shot_recommendations(
4748 project_id : Optional [str ] = None ,
4849) -> List [Dict [str , str ]]:
4950 recommendations = zs_service .get_recommended_models ()
50- existing_models = model_manager .get_model_provider_info ()
51+ if misc .check_is_managed :
52+ existing_models = model_manager .get_model_provider_info ()
53+ else :
54+ existing_models = []
5155
5256 for model in existing_models :
5357 if model ["zero_shot_pipeline" ]:
You can’t perform that action at this time.
0 commit comments