fix: update import

Author: mesakhcienet

src/MaxText/layers/nnx_decoders.py

@@ -29,33 +29,33 @@
 
 from MaxText import max_logging, maxtext_utils, multimodal_utils, sharding
 from MaxText.common_types import (
-  EP_AS_CONTEXT,
-  MODEL_MODE_AUTOREGRESSIVE,
-  MODEL_MODE_PREFILL,
-  MODEL_MODE_TRAIN,
-  Config,
-  DecoderBlockType,
-  ShardMode,
+    EP_AS_CONTEXT,
+    MODEL_MODE_AUTOREGRESSIVE,
+    MODEL_MODE_PREFILL,
+    MODEL_MODE_TRAIN,
+    Config,
+    DecoderBlockType,
+    ShardMode,
 )
 from MaxText.inference import page_manager
 from MaxText.layers import (
-  deepseek,
-  deepseek_batchsplit,
-  gemma,
-  gemma2,
-  gemma3,
-  gpt3,
-  gpt_oss,
-  initializers,
-  linears,
-  llama2,
-  llama4,
-  mistral,
-  mixtral,
-  nnx_wrappers,
-  quantizations,
-  qwen3,
-  simple_layer,
+    deepseek,
+    deepseek_batchsplit,
+    gemma,
+    gemma2,
+    gemma3,
+    gpt3,
+    gpt_oss,
+    initializers,
+    linears,
+    llama2,
+    llama4,
+    mistral,
+    mixtral,
+    nnx_wrappers,
+    quantizations,
+    qwen3,
+    simple_layer,
 )
 from MaxText.layers import nnx_pipeline as pipeline