h|SrSSKJr SSKrSSKrSSKJr SSKJr SS/r "SS5r \RS 5r g) zMiscellaneous utilities.) annotationsN)Callable)EnvCloudpickleWrapperclear_mpi_env_varsc4\rSrSrSrS SjrSrSrSrSr g) rz@Wrapper that uses cloudpickle to pickle and unpickle the result.cXlg)z#Cloudpickle wrapper for a function.Nfn)selfr s U/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/vector/utils/misc.py__init__CloudpickleWrapper.__init__sc@SSKnURUR5$)z1Get the state using `cloudpickle.dumps(self.fn)`.rN) cloudpickledumpsr )r rs r __getstate__CloudpickleWrapper.__getstate__s  ))rc8SSKnURU5Ulg)zSets the state with obs.rN)pickleloadsr )r obrs r __setstate__CloudpickleWrapper.__setstate__s,,r"rc"UR5$)z/Calls the function `self.fn` with no arguments.r )r s r__call__CloudpickleWrapper.__call__"swwyrr N)r zCallable[[], Env]) __name__ __module__ __qualname____firstlineno____doc__rrrr__static_attributes__rrrrsJ* # rc#l# 0n[[RR55H;upSH0nUR U5(dMX U'[RU M2 M= Sv [RR U5 g![RR U5 f=f7f)aClears the MPI of environment variables. ``from mpi4py import MPI`` will call ``MPI_Init`` by default. If the child process has MPI environment variables, MPI will think that the child process is an MPI process just like the parent and do bad things such as hang. This context manager is a hacky way to clear those environment variables temporarily such as when we are starting multiprocessing Processes. Yields: Yields for the context manager )OMPI_PMI_N)listosenvironitems startswithupdate)removed_environmentkvprefixs rrr'sRZZ%%'('F||F##)*A&JJqM() /  -. -.s$AB4B4,B0 B4!B11B4) r$ __future__r contextlibr+collections.abcrgymnasium.corer__all__rcontextmanagerrr&rrr:sH" $ !5 60 //r