hubai_sdk.utils.nn_archive

module documentation

Undocumented

Function cleanup_extracted_path Delete an extracted NN archive path if it exists.
Function process_nn_archive Extracts the archive from tar and parses its config.
def cleanup_extracted_path(path: Path):

Delete an extracted NN archive path if it exists.

def process_nn_archive(path: Path, overrides: dict[str, Any] | None) -> tuple[Config, NNArchiveConfig, str]:

Extracts the archive from tar and parses its config.

Parameters
path:PathPath to the archive.
overrides:dict[str, Any] | NoneConfig overrides.
Returns
tuple[Config, NNArchiveConfig, str]Tuple of the parsed config, NNArchiveConfig, and the main stage key.