hSSrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK r SSK J r J r J r SSKrSSKrSSKrSSKJr SSKJrJr S\ S\S\ 4S jrS\ S\S \ SS4S jrS \ S\4S jrS\\\ 4S\4SjrS%S\S\ \\\ 4S\\\ 44Sjjr\R@S&S\ \\RB\RD4S\S\#S\ \S\ \RH\RJ\RL\RN44 Sjj5r(\(RS\5S&S\S\S\#S\ \S\RD4 Sjj5r*\(RS\RB5S&S\RBS\S\#S\ \S\RD4 Sjj5r+S'S\ \\RB\RD4S\ \\\ 4S\ \\\ 4S\ \\\ 4S\#SS4 Sjjr,S(S\ \\RB\RD4S\ S\#SS4Sjjr-S(S\ \\RB\RD4S\#S\ 4Sjjr.S)S \ \\RB\RD4S!\S\ \\\ 4S"\ \R^\4S\#S#\S\0\ \\\ 4\\ \44S$jjr1g)*zb Save util taken from stable_baselines used to serialize data (class parameters) of model classes N)AnyOptionalUnion) TensorDict) get_deviceget_system_infoobjattrreturnc^S[S[S[4U4Sjjn[R"X0/UR S5Q5$)a Recursive version of getattr taken from https://stackoverflow.com/questions/31174295 Ex: > MyObject.sub_object = SubObject(name='test') > recursive_getattr(MyObject, 'sub_object.name') # return test :param obj: :param attr: Attribute to retrieve :return: The attribute r r r c>[X/TQ76$N)getattr)r r argss \/home/james-whalen/.local/lib/python3.13/site-packages/stable_baselines3/common/save_util.py_getattr#recursive_getattr.._getattr&ss(4((.)rstr functoolsreducesplit)r r rrs ` rrecursive_getattrrs@)c)))   H&=TZZ_&= >>rvalcpURS5up4n[U(a[X5XR5$UXR5$)a Recursive version of setattr taken from https://stackoverflow.com/questions/31174295 Ex: > MyObject.sub_object = SubObject(name='test') > recursive_setattr(MyObject, 'sub_object.name', 'hello') :param obj: :param attr: Attribute to set :param val: New value of the attribute r) rpartitionsetattrr)r r rpre_posts rrecursive_setattrr",s5??3'LCD #$S. JJ3 JJritemc^Sn[R"U5nU$![a SnU$f=f)z Test if an object is serializable into JSON :param item: The object to be tested for JSON serialization. :return: True if object is JSON serializable, false otherwise. TF)jsondumps TypeError)r#json_serializabler s ris_json_serializabler)<sB" JJt   "! "s  ,,datacZ0nUR5Hup#[U5(aX1U'M[R"[R "U55R 5n[[U55US.n[US5(d[U[5(aj[U[5(a UROURRnU"5H(upx[U5(aXU'M[U5XW'M* XQU'GM [R "USS9n U $)a  Turn data (class parameters) into a JSON string for storing :param data: Dictionary of class parameters to be stored. Items that are not JSON serializable will be pickled with Cloudpickle and stored as bytearray in the JSON file :return: JSON string of the data serialized. )z:type: :serialized:__dict__)indent)itemsr)base64 b64encode cloudpickler&decodertypehasattr isinstancedictr-r%) r*serializable_datadata_key data_itembase64_encodedcloudpickle_serializationitem_generator variable_name variable_item json_strings r data_to_jsonrBLs#zz|  * **3h '$--k.?.? .JKRRTN d9o. .) %y*--It1L1L4>y$4O4OU^UgUgUmUm4B4D0M,M::CP-@CF}CU1@ 5E+Dh 'K ,L**.q9K rrAcustom_objectscNUb [U[5(d [S5e[R"U5n0nUR 5HupEUbXAR 5;aXX4'M#[U[5(aZSUR 5;aFUSn[R"UR55n[R"U5nXU'MXSU'M U$![[[4a'n [R"SUSU 35 Sn A MSn A ff=f)a Turn JSON serialization of class-parameters back into dictionary. :param json_string: JSON serialization of the class-parameters that should be loaded. :param custom_objects: Dictionary of objects to replace upon loading. If a variable is present in this dictionary as a key, it will not be deserialized and the corresponding item will be used instead. Similar to custom_objects in ``keras.models.load_model``. Useful when you have an object in file that can not be deserialized. :return: Loaded class parameters. Nz.custom_objects argument must be a dict or Noner,zCould not deserialize object zN. Consider using `custom_objects` argument to replace this object. Exception: )r7r8 ValueErrorr%loadsr0keysr1 b64decodeencoder3 RuntimeErrorr'AttributeErrorwarningswarn) rArC json_dict return_datar:r; serialization base64_objectdeserialized_objectes r json_to_datarTs!*^T*J*JIJJ ;'IK(0  %(6I6I6K*K%3$"#$& s:C((D$=DD$pathmodeverbosesuffixc[R[R[R[R4n[ X5(d[ SUSUS35eUR(a[SUS35eUR5nSSSSS.UnSU:XaUR5(aSU:Xa.UR5(dSU:XaS OS n[S US35eU$![an[S 5UeS nAff=f)as Opens a path for reading or writing with a preferred suffix and raises debug information. If the provided path is a derivative of io.BufferedIOBase it ensures that the file matches the provided mode, i.e. If the mode is read ("r", "read") it checks that the path is readable. If the mode is write ("w", "write") it checks that the file is writable. If the provided path is a string or a pathlib.Path, it ensures that it exists. If the mode is "read" it checks that it exists, if it doesn't exist it attempts to read path.suffix if a suffix is provided. If the mode is "write" and the path does not exist, it creates all the parent folders. If the path points to a folder, it changes the path to path_2. If the path already exists and verbose >= 2, it raises a warning. :param path: the path to open. if save_path is a str or pathlib.Path and mode is "w", single dispatch ensures that the path actually exists. If path is a io.BufferedIOBase the path exists. :param mode: how to open the file. "w"|"write" for writing, "r"|"read" for reading. :param verbose: Verbosity level: 0 for no output, 1 for info messages, 2 for debug messages :param suffix: The preferred suffix. If mode is "w" then the opened file has the suffix. If mode is "r" then we attempt to open the path. If an error is raised and the suffix is not None, we attempt to open the path with the suffix. :return: zPath z- parameter has invalid type: expected one of rz File stream z is closed.wr)writereadrZr[&Expected mode to be either 'w' or 'r'.Nwritablereadablez Expected a z file.) ioBufferedWriterBufferedReaderBytesIOBufferedRandomr7r'closedrElowerKeyErrorr_r`)rUrVrWrX allowed_typesrS error_msgs r open_pathrks8&&(9(92::rGXGXYM d * *%v%RS`Raabcdd {{<v[9:: ::tD dmmooC4K"%+J: ;yk899 K JABIJs C-- D7 DDcD[[R"U5XU5$)a Open a path given by a string. If writing to the path, the function ensures that the path exists. :param path: the path to open. If mode is "w" then it ensures that the path exists by creating the necessary folders and renaming path if it points to a folder. :param mode: how to open the file. "w" for writing, "r" for reading. :param verbose: Verbosity level: 0 for no output, 1 for info messages, 2 for debug messages :param suffix: The preferred suffix. If mode is "w" then the opened file has the suffix. If mode is "r" then we attempt to open the path. If an error is raised and the suffix is not None, we attempt to open the path with the suffix. :return: )open_path_pathlibpathlibPath)rUrVrWrXs r open_path_strrps W\\$/ GGrcUS;a [S5eUS:Xa[URS5XU5$URS:Xa$Ub!US:wa[R "USU35nUR5(a5UR5(a US:a[ R"S US 35 [URS 5XU5$![aVnUbLUS:waF[R "USU35nUS:a[ R"S US US35 USp0SnAOUeSnAff=f![a9 [ R"S US US35 [R "US35nOM[aA [ R"S URS35 URRSSS9 Of=f[XX#5$)a Open a path given by a string. If writing to the path, the function ensures that the path exists. :param path: the path to check. If mode is "w" then it ensures that the path exists by creating the necessary folders and renaming path if it points to a folder. :param mode: how to open the file. "w" for writing, "r" for reading. :param verbose: Verbosity level: 0 for no output, 2 for indicating if path without suffix is not found when mode is "r" :param suffix: The preferred suffix. If mode is "w" then the opened file has the suffix. If mode is "r" then we attempt to open the path. If an error is raised and the suffix is not None, we attempt to open the path with the suffix. :return: )rZr[r^r[rbNrzPath 'z' not found. Attempting z' exists, will overwrite it.wbz$' is a folder. Will save instead to _2z!' does not exist. Will create it.T)exist_okparents)rErkopenFileNotFoundErrorrnrorLrMrXexistsis_fileIsADirectoryErrorparentmkdirrm)rUrVrWrXerrornewpaths rrmrms  :ABB s{ TYYt_dVD D ;{{b V%7FbL||tfAfX$67{{}}GqL tf,HIJTYYt_dVD D! !fl!,,$q'9:a<MMF4&0H QR"ST&f  ! - MMF4&(LTFRTU V<<4& ,D  ; MMF4;;-/PQ R KK  tT  : ; T 99s8CBD3 D0A D+)D++D03AF?5AF?>F? save_pathparamspytorch_variablesc[USSSS9nUb [U5n[R"USS9nUbUR SW5 Ub0UR SSS S 9n[ R"X85 SSS5 UbLUR5H8upUR U S -SS S 9n [ R"X5 SSS5 M: UR S [R5 UR S [SS9S5 SSS5 [U[[R45(aUR!5 gg!,(df  N=f!,(df  M=f!,(df  Nh=f)a6 Save model data to a zip archive. :param save_path: Where to store the model. if save_path is a str or pathlib.Path ensures that the path actually exists. :param data: Class parameters being stored (non-PyTorch variables) :param params: Model parameters being stored expected to contain an entry for every state_dict with its name and the state_dict. :param pytorch_variables: Other PyTorch variables expected to contain name and value of the variable. :param verbose: Verbosity level: 0 for no output, 1 for info messages, 2 for debug messages rZrziprWrXNrVr*pytorch_variables.pthT)rV force_zip64.pth_stable_baselines3_versionsystem_info.txtF) print_info)rkrBzipfileZipFilewritestrrythsaver0sb3 __version__rr7rrnroclose) rr*rrrWfileserialized_dataarchivepytorch_variables_file file_namedict_ param_files rsave_to_zip_filer&s?$ YQu =D &t, C (G     V_ 5  (5CTRVl)BS  $*LLN \\)f"43D\QU_GGE.RQ%3 5sG*Ou,Ma,PQ ) )c7<<011 2SRRQ ) (s<*E&E14E&%ED KKF$;$;<$gll+,, -rc[USUSS9n[R"U5n[U[[ R 45(aUR5 U$)a Load an object from the path. If a suffix is provided in the path, it will use that suffix. If the path does not exist, it will attempt to load using the .pkl suffix. :param path: the path to open. if save_path is a str or pathlib.Path and mode is "w", single dispatch ensures that the path actually exists. If path is a io.BufferedIOBase the path exists. :param verbose: Verbosity level: 0 for no output, 1 for info messages, 2 for debug messages r[rr)rkrloadr7rrnror)rUrWrr s r load_from_pklrgsF T3 >D ++d C$gll+,, Jr load_path load_datadeviceprint_system_infoc $[USUSS9n[US9n[R"U5nUR 5nSn Sn 0n U(aUSU;a4[ S5 [ UR S5R55 O[R"S[5 S U;a/U(a(UR S 5R5n [XS 9n UV s/sH-n [RRU 5S S :XdM+U PM/ nn UHnURUSS 9n[ R""5nUR%UR 55 UR'S5 [(R*"UUSS9nUS:XdUS:XaUn O%UU [RRU5S'SSS5 M SSS5 [1U[2[4R645(aUR95 W W W 4$s sn f!,(df  GM=f!,(df  N`=f![R,an[/SUS35UeSnAff=f![1U[2[4R645(aUR95 ff=f)a Load model data from a .zip archive :param load_path: Where to load the model from :param load_data: Whether we should load and return data (class parameters). Mainly used by 'load_parameters' to only load model parameters (weights) :param custom_objects: Dictionary of objects to replace upon loading. If a variable is present in this dictionary as a key, it will not be deserialized and the corresponding item will be used instead. Similar to custom_objects in ``keras.models.load_model``. Useful when you have an object in file that can not be deserialized. :param device: Device on which the code should run. :param verbose: Verbosity level: 0 for no output, 1 for info messages, 2 for debug messages :param print_system_info: Whether to print or not the system info about the saved model. :return: Class parameters, model state_dicts (aka "params", dict of state_dict) and dict of pytorch variables r[rr)rNrz== SAVED MODEL SYSTEM INFO ==zOThe model was saved with SB3 <= 1.2.0 and thus cannot print system information.r*)rCrrrrF) map_location weights_onlyrz tensors.pthzError: the file z wasn't a zip-file)rkrrrnamelistprintr]r4rLrM UserWarningrTosrUsplitextryrardr\seekrr BadZipFilerEr7rrnror)rrrCrrWrrrrr*rr json_datar pth_files file_pathr file_content th_objectrSs rload_from_zip_filerxs?6 YWU CDv &F8 __T "g'')HD $ F!$09:',,'89@@BCMMi# !i$LL0779 #IM 5=iHy@P@PQZ@[\]@^bh@hHIi& \\)#\6*$&::%76'=#l i#w||!4 5 5 JJL * **7j66?# "d   R+I;6HIJPQQR i#w||!4 5 5 JJL 6sxH+B,H*HH H$BH4 HH+H H  H H($H+'I(H++I?III8Jr)rN)NNNr)r)TNautorF)2__doc__r1rrar%rrnrrLrtypingrrrr3torchrstable_baselines3r%stable_baselines3.common.type_aliasesrstable_baselines3.common.utilsrrrrr"boolr)r8rBrTsingledispatchroBufferedIOBaseintrbrcrdrerkregisterrprmrrrrtuplerrrrs   ''<F?3?c?S?& K3 Kc K K K s t  4tCH~4#4n/c/8DcN3K/W[\_ad\dWe/d lp( W\\2#4#44 5(=@(KN(\deh\i( 2  b//R=N=N NO((V CHH3HH(3-H[][l[lHH" GLL!/:GLL/:/:c/:xX[}/:hjhyhy/:"/:h&*'+26 *S',,(9(99:* 4S> "* T#s(^ $* S#X/ *  *  *ZeCr/@/@@AVYbf(c7<<1B1BBCcZ]&/3$*# Z+S',,(9(99:Z+Z+T#s(^,Z+ "))S. ! Z+  Z+  Z+ 8DcN #Z*1E EFZ+r