pybaum.registry_entries#

Module Contents#

Functions#

_none()

Create registry entry for NoneType.

_list()

Create registry entry for list.

_dict()

Create registry entry for dict.

_tuple()

Create registry entry for tuple.

_namedtuple()

Create registry entry for namedtuple and NamedTuple.

_unflatten_namedtuple(aux_data, leaves)

_ordereddict()

Create registry entry for OrderedDict.

_numpy_array()

Create registry entry for numpy.ndarray.

_array_element_names(arr)

_jax_array()

_pandas_series()

Create registry entry for pandas.Series.

_pandas_dataframe()

Create registry entry for pandas.DataFrame.

_flatten_pandas_dataframe(df)

_unflatten_pandas_dataframe(aux_data, leaves)

_get_names_pandas_dataframe(df)

_index_element_to_string(element)

Attributes#

FUNC_DICT

pybaum.registry_entries._none()[source]#

Create registry entry for NoneType.

pybaum.registry_entries._list()[source]#

Create registry entry for list.

pybaum.registry_entries._dict()[source]#

Create registry entry for dict.

pybaum.registry_entries._tuple()[source]#

Create registry entry for tuple.

pybaum.registry_entries._namedtuple()[source]#

Create registry entry for namedtuple and NamedTuple.

pybaum.registry_entries._unflatten_namedtuple(aux_data, leaves)[source]#
pybaum.registry_entries._ordereddict()[source]#

Create registry entry for OrderedDict.

pybaum.registry_entries._numpy_array()[source]#

Create registry entry for numpy.ndarray.

pybaum.registry_entries._array_element_names(arr)[source]#
pybaum.registry_entries._jax_array()[source]#
pybaum.registry_entries._pandas_series()[source]#

Create registry entry for pandas.Series.

pybaum.registry_entries._pandas_dataframe()[source]#

Create registry entry for pandas.DataFrame.

pybaum.registry_entries._flatten_pandas_dataframe(df)[source]#
pybaum.registry_entries._unflatten_pandas_dataframe(aux_data, leaves)[source]#
pybaum.registry_entries._get_names_pandas_dataframe(df)[source]#
pybaum.registry_entries._index_element_to_string(element)[source]#
pybaum.registry_entries.FUNC_DICT[source]#