o xi'@sddlZddlZddlmZddlZddlmZdZGdddZde d dfd d Z de d dfd d Z de ded dfddZ de d efddZ deded dfddZdS)N)Any) LaunchErrorWANDB_LAUNCH_FILE_OVERRIDESc@s&eZdZdZdZddZdddZdS) FileOverrideszCSingleton that read file overrides json from environment variables.NcCs.|jdurt||_i|j_|j|jSN) _instanceobject__new__ overridesload)clsrQ/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/launch/inputs/files.pyr s   zFileOverrides.__new__returncCstjt}|dur:tdtjvr:d}d}td|tjvr:|tjtd|7}|d7}td|tjvs |rczt|}t|tsNtdt||_ WdStj ybtdtwdS)z,Load overrides from an environment variable.N_0r_zInvalid JSON in ) osenvirongetFILE_OVERRIDE_ENV_VARjsonloads isinstancedictrr JSONDecodeError)selfr idxcontentsrrrr s$    zFileOverrides.load)rN)__name__ __module__ __qualname____doc__rr r rrrrr s rpathrcstjrtdddvrtddtjtjs.tddtfdddDsAtdd d S) aValidate a config file path. This function checks if a given config file path is valid. A valid path should meet the following criteria: - The path must be expressed as a relative path without any upwards path traversal, e.g. `../config.json`. - The file specified by the path must exist. - The file must have a supported extension (`.json`, `.yaml`, or `.yml`). Args: path (str): The path to validate. Raises: LaunchError: If the path is not valid. zInvalid config path: z!. Please provide a relative path.z..zZ. Please provide a relative path without any upward path traversal, e.g. `../config.json`.z. File does not exist.c3s|]}|VqdSr)endswith).0extr%rr Hsz'config_path_is_valid..).json.yaml.ymlz). Only JSON and YAML files are supported.N)rr%isabsrnormpathexistsanyr)rr)rconfig_path_is_valid+s      r2cCsLt}||jvr"|j|}|dur$t|}t||t||dSdSdS)zDCheck for file overrides in the environment and apply them if found.N)rr r_read_config_file _update_dict_write_config_file)r%file_overridesr configrrr override_fileNs   r8r7cCstj|\}}|dkr,t|d}tj||ddWddS1s%wYdS|dvrNt|d}t||WddS1sGwYdStd|)zWrite a config file to disk. Args: path (str): The path to the config file. config (Any): The contents of the config file as a Python object. Raises: LaunchError: If the file extension is not supported. r+wr)indentNr,r-Unsupported file extension: ) rr%splitextopenrdumpyaml safe_dumpr)r%r7rr(frrrr5Ys  " "r5cCstj|\}}|dkr't| }t|WdS1s wYdS|dvrFt| }t|WdS1s?wYdStd|)zRead a config file from disk. Args: path (str): The path to the config file. Returns: Any: The contents of the config file as a Python object. r+Nr;r<) rr%r=r>rr r@ safe_loadr)r%rr(rBrrrr3ns  $$r3targetsourcecCsH|D]\}}t|tr||vri||<t|||q|||<qdS)zUpdate a dictionary with the contents of another dictionary. Args: target (Dict): The dictionary to update. source (Dict): The dictionary to update from. N)itemsrrr4)rDrEkeyvaluerrrr4s  r4)rrtypingrr@errorsrrrstrr2r8r5r3rr4rrrrs  #