noether.data.datasets.cfd.emmi_wing¶

Submodules¶

Classes¶

`EmmiWingDataset`	Dataset implementation for aerodynamic datasets with volume and surface fields.
`EmmiWingHFDataset`	Emmi-Wing dataset loaded from the HuggingFace subset.

Package Contents¶

class noether.data.datasets.cfd.emmi_wing.EmmiWingDataset(dataset_config)¶

Bases: noether.data.datasets.cfd.dataset.AeroDataset

Dataset implementation for aerodynamic datasets with volume and surface fields. This unified dataset class provides an interface for aerodynamics dataset with volume and surface fields. The dataset behavior such as the dataset choice, train/val/test split IDs, etc. is configured through constructor parameters, allowing for easy extension to new datasets.

Parameters:

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

STATS_FILE: str = ''¶

split¶

source_root¶

sample_info(idx)¶

Get information about a sample such as its local path, run name, etc.

Parameters:: idx (int)
Return type:: dict[str, str | int | None]

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]

getitem_geometry_design_parameters(idx)¶

Retrieves geometry design parameters as a single tensor.

Returns:: Geometry design parameters tensor of shape (1, num_geometry_parameters)
Return type:: torch.Tensor
Parameters:: idx (int)

getitem_inflow_design_parameters(idx)¶

Retrieves inflow design parameters as a single tensor.

Returns:: Inflow design parameters tensor of shape (1, num_inflow_parameters)
Return type:: torch.Tensor
Parameters:: idx (int)

class noether.data.datasets.cfd.emmi_wing.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.core.schemas.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.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