From ffcdb793b5c14dae2ab62bb531b401dc60113e2c Mon Sep 17 00:00:00 2001 From: Dipannita Shaw Date: Fri, 20 Feb 2026 23:43:56 +0000 Subject: [PATCH] Fix Jupter NB Test failure --- src/maxtext/integration/tunix/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/maxtext/integration/tunix/utils.py b/src/maxtext/integration/tunix/utils.py index 454583d4b..28d9e8377 100644 --- a/src/maxtext/integration/tunix/utils.py +++ b/src/maxtext/integration/tunix/utils.py @@ -16,7 +16,7 @@ import re -import MaxText.integration.tunix.weight_mapping as weight_mapping # pylint: disable=consider-using-from-import +import maxtext.integration.tunix.weight_mapping as weight_mapping # pylint: disable=consider-using-from-import from maxtext.checkpoint_conversion.utils.param_mapping import PARAM_MAPPING from maxtext.checkpoint_conversion.utils.param_mapping import VLLM_HOOK_FNS