id_translation.dio.integration.dask

Integration for Dask types.

Module Attributes

DaskT

Supported dask types.

Classes

DaskIO()

IO implementation for dask types.

class DaskIO

Bases: DataStructureIO[DaskT, str, SourceType, IdType]

IO implementation for dask types.

classmethod extract(translatable, names)

Extract IDs from translatable.

Parameters:

• translatable – Data to extract IDs from.

• names – List of names in translatable to extract IDs for.

Returns:

A dict {name: ids}.

classmethod handles_type(arg)

Return True if the implementation handles data for the type of arg.

classmethod insert(translatable, names, tmap, copy)

Insert translations into translatable.

Parameters:

• translatable – Data to translate. Modified iff copy=False.

• names – Names in translatable to translate.

• tmap – Translations for IDs in translatable.

• copy – If True, modify contents of the original translatable. Otherwise, returns a copy.

Returns:

A copy of translatable if copy=True, None otherwise.

Raises:

NotInplaceTranslatableError – If copy=False for a type which is not translatable in-place.

classmethod names(translatable)

Extract names from translatable.

Parameters:

translatable – Data to extract names from.

Returns:

A list of names to translate. Returns None if names cannot be extracted.

class DaskT

Supported dask types.

alias of TypeVar(‘DaskT’, ~dask_expr._collection.DataFrame, ~dask_expr._collection.Series)