diff --git a/src/ptbench/data/typing.py b/src/ptbench/data/typing.py index 344c1294df6777f88acdde23dec51d40fc51e31e..e8dc56a02c55687ecebefd84d2ab8c27e23cafc6 100644 --- a/src/ptbench/data/typing.py +++ b/src/ptbench/data/typing.py @@ -4,6 +4,7 @@ """Defines most common types used in code.""" +import collections.abc import typing import torch @@ -47,12 +48,11 @@ Typically used in data-processing pipelines inside pytorch. TransformSequence = typing.Sequence[Transform] """A sequence of transforms.""" -DatabaseSplit = dict[str, typing.Sequence[typing.Any]] -"""The definition of a database script. +DatabaseSplit = collections.abc.Mapping[str, typing.Sequence[typing.Any]] +"""The definition of a database split. -A database script maps subset names to sequences of objects that, -through RawDataLoader's eventually become Samples in the processing -pipeline. +A database split maps subset names to sequences of objects that, through +RawDataLoader's eventually become Samples in the processing pipeline. """