def save_job(cls, job: Job, *, module=None, module_data=None) -> NoReturn:
"""
Save job
:param job: job to save
:param module: job module
:param module_data:
"""
job_path = JobPath(job.job_id)
job_path.makedirs()
cls.__save_json(job_path.metadata_file, job.metadata)
cls.__save_json(job_path.job_config_file, job.job_config)
cls.__save_json(job_path.train_config_file, job.train_config)
cls.__save_json(job_path.aggregate_config_file, job.aggregate_config)
if module_data is not None:
ZipUtils.extract_data(module_data, GflConf.temp_dir)
ModuleUtils.migrate_module(
PathUtils.join(GflConf.temp_dir, job.job_id),
job_path.module_name, job_path.module_dir)
elif module is not None:
ModuleUtils.submit_module(module, job_path.module_name,
job_path.module_dir)
else:
ModuleUtils.submit_module(job.module, job_path.module_name,
job_path.module_dir)
def save_dataset(cls,
dataset: Dataset,
*,
module=None,
module_data=None) -> NoReturn:
"""
Save dataset
:param dataset: dataset to save
:param module: dataset module
"""
if module is None:
module = dataset.module
dataset_path = DatasetPath(dataset.dataset_id)
dataset_path.makedirs()
cls.__save_json(dataset_path.metadata_file, dataset.metadata)
cls.__save_json(dataset_path.dataset_config_file,
dataset.dataset_config)
if module_data is not None:
ZipUtils.extract_data(module_data, GflConf.temp_dir)
ModuleUtils.migrate_module(
PathUtils.join(GflConf.temp_dir, dataset.dataset_id),
dataset_path.module_name, dataset_path.module_dir)
elif module is not None:
ModuleUtils.submit_module(module, dataset_path.module_name,
dataset_path.module_dir)
else:
ModuleUtils.submit_module(dataset.module, dataset_path.module_name,
dataset_path.module_dir)
from gfl.utils import ZipUtils
if __name__ == "__main__":
data = ZipUtils.get_compress_data("__init__.py", "")
ZipUtils.extract_data(data, "t2")