o niM@sddlZddlZddlmZmZddlZddlmZdZGdddZ de d dfd d Z de d dfd d Z de ded dfddZ de d efddZdeded dfddZdS)N)AnyDict) LaunchErrorWANDB_LAUNCH_FILE_OVERRIDESc@s&eZdZdZdZddZdddZdS) FileOverrideszCSingleton that read file overrides json from environment variables.NcCs.|jdurt||_i|j_|j|jSN) _instanceobject__new__ overridesload)clsr[/home/ubuntu/SoloSpeech/.venv/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__r r 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 Isz'config_path_is_valid..).json.yaml.ymlz). Only JSON and YAML files are supported.N)rr&isabsrnormpathexistsanyr*rr*rconfig_path_is_valid,s      r3cCsLt}||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_fileOs   r9r8cCstj|\}}|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&r8rr)frrrr6Zs  " "r6cCstj|\}}|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 rA safe_loadr)r&rr)rCrrrr4os  $$r4targetsourcecCsH|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)itemsrrr5)rErFkeyvaluerrrr5s  r5)rrtypingrrrAerrorsrrrstrr3r9r6r4r5rrrrs  #