From aefd70a2b3772d91c10ef5a1c104bb000919c671 Mon Sep 17 00:00:00 2001
From: Igor Tsvetkov <igorts@google.com>
Date: Mon, 16 Mar 2026 16:10:58 -0700
Subject: [PATCH] tmp

---
 src/maxtext/common/checkpointing.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/maxtext/common/checkpointing.py b/src/maxtext/common/checkpointing.py
index 220ff6f16d..769a15399b 100644
--- a/src/maxtext/common/checkpointing.py
+++ b/src/maxtext/common/checkpointing.py
@@ -16,6 +16,7 @@
 
 import time
 from typing import Any, Optional
+from packaging import version
 
 from absl import flags
 import datetime
@@ -774,7 +775,7 @@ def save_checkpoint(checkpoint_manager, step, state, config=None, data_iterator=
       ocdbt_target_data_file_size=chunk_byte_size,
   )
   save_args_composite = {"items": checkpoint_args}
-
+  print(version.__package__)
   if (
       config
       and config.dataset_type == "grain"