diff --git a/MaxText/train.py b/MaxText/train.py
index 3c1b3ac07..0d35c4d99 100644
--- a/MaxText/train.py
+++ b/MaxText/train.py
@@ -44,6 +44,7 @@
 # pylint: disable-next=unused-import
 import register_jax_proxy_backend
 from vertex_tensorboard import VertexTensorboardManager
+# Placeholder: internal
 
 from input_pipeline.input_pipeline_interface import create_data_iterator_with_tokenizer
 from layers import models