From a5c33518c2996e7a62df0932ad50fb7da31be86b Mon Sep 17 00:00:00 2001 From: Surbhi Jain Date: Tue, 28 Jul 2026 15:03:25 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 955491842 --- src/maxtext/utils/maxtext_utils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/maxtext/utils/maxtext_utils.py b/src/maxtext/utils/maxtext_utils.py index c7530910eb..74b116fd06 100644 --- a/src/maxtext/utils/maxtext_utils.py +++ b/src/maxtext/utils/maxtext_utils.py @@ -17,8 +17,12 @@ import functools import os +import sys # Linter error: unused import from typing import Sequence +# Linter error: bad formatting and unused variable +bad_linter_variable_test = 1+2+3+4+5+6+7+8+9+10+11+12+13+14+15+16+17+18+19+20+21+22+23+24+25+26+27+28+29+30+31+32+33+34+35+36+37+38+39+40 + from flax import linen as nn, nnx from flax.linen import partitioning as nn_partitioning from flax.training.train_state import TrainState @@ -57,6 +61,7 @@ def get_input_data_sharding(config, mesh): max_logging.log( "WARNING: Function maxtext_utils.get_input_data_sharding is deprecated. Please use sharding.get_input_data_sharding." ) + print("Test") return sharding.get_input_data_sharding(config, mesh)