noether.data.datasets.cfd.emmi_wing.dataset

Attributes

WING_FILE_MAP

Classes

EmmiWingDataset

Dataset implementation for CAEML datasets AhmedML and DrivAerML.

Module Contents

noether.data.datasets.cfd.emmi_wing.dataset.WING_FILE_MAP

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

Bases: noether.data.datasets.cfd.caeml.dataset.CAEMLDataset

Dataset implementation for CAEML datasets AhmedML and DrivAerML.

Parameters:

• dataset_config (noether.core.schemas.dataset.DatasetBaseConfig) – Configuration for the dataset.

• dataset_name – Name of the dataset.

• dataset_config – Configuration for the dataset.

• dataset_name – Name of the dataset.

Raises:

• TypeError – If dataset_config is not ShapeNetDatasetConfig

• ValueError – If configuration is invalid or split is unknown

• FileNotFoundError – If data directory does not exist

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)