noether.data.datasets.cfd.emmi_wing.dataset_hf

Attributes

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:
property get_dataset_splits: noether.core.schemas.dataset.DatasetSplitIDs
Return type:

noether.core.schemas.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