morpheus.utils.type_aliases

Type aliases used throughout the codebase.

DataFrameModule[source]

Valid DataFrame modules.

alias of Literal[‘cudf’, ‘pandas’]

DataFrameType[source]

Alias for pandas and cuDF DataFrame types.

alias of Union[pandas.DataFrame, cudf.DataFrame]

NDArrayType[source]

Alias for NumPy and CuPy ndarray types.

alias of Union[numpy.ndarray, cupy.ndarray]

SeriesType[source]

Alias for pandas and cuDF Series types.

alias of Union[pandas.Series, cudf.Series]

TensorMapType[source]

Alias for a dictionary of tensor names to tensors represented as either a NumPy or CuPy ndarray.

alias of Dict[str, Union[numpy.ndarray, cupy.ndarray]]