From fe4658ebd0d2ba66ce6424bc18fdfc18b9ee2510 Mon Sep 17 00:00:00 2001 From: Roman Kryvokhyzha Date: Fri, 20 Feb 2026 14:37:44 +0200 Subject: [PATCH] fix: remove unused import of maxtext_google from train.py --- src/maxtext/trainers/pre_train/train.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/maxtext/trainers/pre_train/train.py b/src/maxtext/trainers/pre_train/train.py index 640ce56883..6bb4e46d5a 100644 --- a/src/maxtext/trainers/pre_train/train.py +++ b/src/maxtext/trainers/pre_train/train.py @@ -42,8 +42,6 @@ from MaxText import sharding from MaxText.common_types import ShardMode from MaxText.globals import EPS -# pylint: disable-next=unused-import -from maxtext import maxtext_google from MaxText.gradient_accumulation import gradient_accumulation_loss_and_grad from MaxText.vocabulary_tiling import vocab_tiling_linen_loss