Skip to content

Change how JAX Keras backend requirement is reported#2420

Merged
chensuyue merged 2 commits intointel:masterfrom
anko-intel:dev/anko/fix_import
Mar 11, 2026
Merged

Change how JAX Keras backend requirement is reported#2420
chensuyue merged 2 commits intointel:masterfrom
anko-intel:dev/anko/fix_import

Conversation

@anko-intel
Copy link
Contributor

Type of Change

bug fix

Description

This PR centralizes the Keras JAX backend requirement check into a single reusable check_backend() function and changes the behavior from crashing at import time to issuing a warning at import time and raising an error only at API entry points (quantize_model and prepare_deserialized_quantized_model).

@anko-intel anko-intel force-pushed the dev/anko/fix_import branch from ca01a4b to 31ebdb3 Compare March 10, 2026 16:04
Signed-off-by: Andrzej Kotłowski <andrzej.kotlowski@intel.com>
Copy link
Contributor

@bkowalskiINTEL bkowalskiINTEL left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@chensuyue chensuyue merged commit 49908d7 into intel:master Mar 11, 2026
14 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants