from immuneML.data_model.dataset.Dataset import Dataset
from immuneML.util.PathBuilder import PathBuilder
from immuneML.workflows.steps.data_splitter.DataSplitterParams import DataSplitterParams
[docs]
class Util:
[docs]
@staticmethod
def prepare_path(input_params: DataSplitterParams, split_index: int) -> str:
path = input_params.paths[split_index] / "datasets"
PathBuilder.build(path)
return path
[docs]
@staticmethod
def make_dataset(dataset: Dataset, indices, input_params: DataSplitterParams, i: int, dataset_type: str):
path = Util.prepare_path(input_params, i)
new_dataset = dataset.make_subset(indices, path, dataset_type)
return new_dataset