Skip to content

Commit 56edff8

Browse files
JWittmeyerJWittmeyer
andauthored
Adds hotfix (#94)
Co-authored-by: JWittmeyer <[email protected]>
1 parent bf7e7f7 commit 56edff8

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

controller/embedding/manager.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,16 @@
55
from util import daemon
66
from . import util
77
from . import connector
8+
from controller.misc import manager as misc
89
from controller.model_provider import manager as model_manager
910

1011

1112
def 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"]:

controller/zero_shot/manager.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
from util import daemon
2424
from controller.weak_supervision import weak_supervision_service as weak_supervision
2525
from controller.model_provider import manager as model_manager
26+
from controller.misc import manager as misc
2627

2728

2829
def 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"]:

0 commit comments

Comments
 (0)