Skip to content

PTransform input_types hint with potential empty input #18774

@kennknowles

Description

@kennknowles

User reported issue:

... PTransform class that takes a pcollection of Dict[str, numpy.ndarray] as input, and I added the annotation of @beam.typehints.with_input_types(beam.typehints.Dict[str, numpy.ndarray]). However, this causes error to empty dict input, and the error message is """Input type hint violation at _TopKStatsGeneratorImpl: expected Dict[str, ndarray], got Dict[Union[], Union[]] """

From the user report it seems like our type hints could be more flexible in what they accept for empty dictionaries.

Imported from Jira BEAM-4506. Original Jira may contain additional context.
Reported by: altay.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions