pi6dZddlmZddlmZdededfdZy)z Utility functions for the experimental checkpoint module. This module contains helper functions and utilities used across the experimental checkpoint functionality. )Future)Anyoriginal_resultreturnNctttr)dttddffd }j|Sj dS)a Wraps a result (Future or not) to return a Future with None result. If the input is a Future, returns a new Future that completes with None when the original Future completes successfully, or propagates any exception. If the input is not a Future, returns a completed Future with None result. Args: original_result: The result to wrap (Future or any other value). Returns: A Future that completes with None on success or propagates exceptions. _rNc jjdy#t$r}j|Yd}~yd}~wwxYw)N)result set_result Exception set_exception)re masked_futurers r/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/torch/distributed/checkpoint/_experimental/utils.py on_completez wrap_future..on_completesC /&&(((. /++A.. /s!% A AA )r isinstanceradd_done_callbackr )rrrs` @r wrap_futurer s[#)(M/6* /6#; /4 / ))+6    & )__doc__concurrent.futuresrtypingrrrrrs)&r