src.driver.data_manager

Data management and output handling for MMM V2 workflow.

This module handles data loading, validation, output directory management, and Colab-specific utilities.

Module Contents

class src.driver.data_manager.DataManager(driver)

Handles data management and output operations for V2 architecture.

This class manages output directory creation, file organization, and Google Colab integration utilities.

init_output(data_dir: str = '.') → str

Initialises and creates the output directory for results.

Parameters:

data_dir – Base directory for results. Defaults to “.”.

Returns:

The absolute path to the created results directory.

create_downloadable_zip(excluded_files: List[str], source_folder: str = '/content', zip_name: str = 'Model_files.zip') → None

Creates a downloadable zip file, excluding specified files. Colab-specific.

Parameters:

• excluded_files – List of file patterns to exclude from zip.

• source_folder – Source directory to zip.

• zip_name – Name of the output zip file.