`i>dZddlmZddlmZdededfdZdS)z Utility functions for the experimental checkpoint module. This module contains helper functions and utilities used across the experimental checkpoint functionality. )Future)Anyoriginal_resultreturnNctttr1dttddffd }|ndS)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 ddS#t$r }|Yd}~dSd}~wwxYw)N)result set_result Exception set_exception)re masked_futurers /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/distributed/checkpoint/_experimental/utils.py on_completez wrap_future..on_completesy /&&(((((..... / / /++A......... /s). AAA)r isinstanceradd_done_callbackr )rrrs` @r wrap_futurer s#)((M/6** ' /6#; /4 / / / / / / / ))+6666   &&& )__doc__concurrent.futuresrtypingrrrrrse&%%%%%r