diff --git a/src/maxtext/training_engine/maxtext_engine.py b/src/maxtext/training_engine/maxtext_engine.py index 0e4afc8842..06f3b56e8d 100644 --- a/src/maxtext/training_engine/maxtext_engine.py +++ b/src/maxtext/training_engine/maxtext_engine.py @@ -329,13 +329,16 @@ def get_metrics(self, clear_cache: bool = True) -> abstract_engine.MetricsBuffer return self._metrics_recorder.get_metrics(clear_cache=clear_cache) def prepare_weight_sync(self, **kwargs: Any) -> Any: - """Stages weights for transfer and returns access coordinates. - + """Returns weight synchronization metadata. + + Weight synchronization is not implemented yet. This placeholder returns an + empty dictionary until a transfer backend is integrated. + Args: - **kwargs: Weight staging parameters. - + **kwargs: Reserved for future weight-staging configuration. + Returns: - Synchronization endpoints or coordinates for rollout actors. + An empty dictionary. """ return {}