noether.data.datasets.cfd.caeml.dataset

Attributes

logger

Classes

CAEMLDataset

Dataset implementation for CAEML datasets AhmedML and DrivAerML.

Module Contents

noether.data.datasets.cfd.caeml.dataset.logger

class noether.data.datasets.cfd.caeml.dataset.CAEMLDataset(dataset_config, dataset_name, filemap=CAEML_FILEMAP)

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

Dataset implementation for CAEML datasets AhmedML and DrivAerML.

Parameters:

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

• dataset_name (str) – 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

split

source_root

property supported_splits: set[str]

Return type:

set[str]

property get_dataset_splits: noether.core.schemas.dataset.DatasetSplitIDs

Abstractmethod:

Return type:

noether.core.schemas.dataset.DatasetSplitIDs

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]