Skip to content

Commit e7d49c5

Browse files
OliverBryantclaude
andcommitted
Fix CI errors
- Add launcher imports to flexible launchers __init__.py - Update import statements in test_llm_family.py to use get_registered_llm_families 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
1 parent 0199099 commit e7d49c5

File tree

2 files changed

+20
-3
lines changed

2 files changed

+20
-3
lines changed

xinference/model/flexible/launchers/__init__.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,3 +11,8 @@
1111
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
14+
15+
from .image_process_launcher import launcher as image_process
16+
from .modelscope_launcher import launcher as modelscope
17+
from .transformers_launcher import launcher as transformers
18+
from .yolo_launcher import launcher as yolo

xinference/model/llm/tests/test_llm_family.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,11 @@ def test_cache_from_uri_local():
207207

208208

209209
def test_custom_llm():
210-
from ..custom import get_user_defined_llm_families, register_llm, unregister_llm
210+
from ..custom import get_registered_llm_families as get_user_defined_llm_families
211+
from ..custom import (
212+
register_llm,
213+
unregister_llm,
214+
)
211215

212216
spec = LlamaCppLLMSpecV2(
213217
model_format="ggufv2",
@@ -239,7 +243,11 @@ def test_custom_llm():
239243

240244
def test_persistent_custom_llm():
241245
from ....constants import XINFERENCE_MODEL_DIR
242-
from ..custom import get_user_defined_llm_families, register_llm, unregister_llm
246+
from ..custom import get_registered_llm_families as get_user_defined_llm_families
247+
from ..custom import (
248+
register_llm,
249+
unregister_llm,
250+
)
243251

244252
spec = LlamaCppLLMSpecV2(
245253
model_format="ggufv2",
@@ -663,7 +671,11 @@ def test_quert_engine_SGLang():
663671

664672

665673
def test_query_engine_general():
666-
from ..custom import get_user_defined_llm_families, register_llm, unregister_llm
674+
from ..custom import get_registered_llm_families as get_user_defined_llm_families
675+
from ..custom import (
676+
register_llm,
677+
unregister_llm,
678+
)
667679
from ..llama_cpp.core import XllamaCppModel
668680
from ..llm_family import LLM_ENGINES, check_engine_by_spec_parameters
669681

0 commit comments

Comments
 (0)