noether.data.datasets.cfd.emmi_wing.dataset_hf

Attributes

logger

HF_REPO_ID

Classes

EmmiWingHFDataset

Emmi-Wing dataset loaded from the HuggingFace subset.

Module Contents

noether.data.datasets.cfd.emmi_wing.dataset_hf.logger
noether.data.datasets.cfd.emmi_wing.dataset_hf.HF_REPO_ID = 'EmmiAI/Emmi-Wing'
class noether.data.datasets.cfd.emmi_wing.dataset_hf.EmmiWingHFDataset(dataset_config)

Bases: noether.data.datasets.cfd.emmi_wing.dataset.EmmiWingDataset

Emmi-Wing dataset loaded from the HuggingFace subset.

Uses the 248-case evaluation scan subset with its own train/val/test splits. The dataset can be auto-downloaded from HuggingFace using download().

Parameters:

  • dataset_config (noether.data.base.dataset.StandardDatasetConfig) – Configuration for the dataset. See DatasetBaseConfig for available options.

  • filemap – FileMap object defining the mapping of data properties to filenames. See FileMap for details.

property get_dataset_splits: noether.data.base.dataset.DatasetSplitIDs
Return type:

noether.data.base.dataset.DatasetSplitIDs

property supported_splits: set[str]
Return type:

set[str]

static download(local_dir)

Download and extract the HF subset to a local directory.

Downloads scans.zip from HuggingFace, extracts the nested run_N.zip archives into <local_dir>/run_N/ directories, and cleans up the zip files.

Parameters:

local_dir (str) – Destination directory.

Returns:

Path to the extracted dataset root.

Return type:

str