Skip to content

Conversation

@copybara-service
Copy link

Add dill support for enp typings

Currently, running some functions on Apache Beam leads to errors.

This can be boiled down to this piece of code not working:

import dill
from etils.enp.typing import ui8
dill.dumps(ui8['N'])
>> PicklingError: Can't pickle ui8[N]: it's not found as etils.enp.array_types.typing.ui8

Additionally, if dill is not available, the behavior doesn't change.

Currently, running some functions on Apache Beam leads to errors.

This can be boiled down to this piece of code not working:
```
import dill
from etils.enp.typing import ui8
dill.dumps(ui8['N'])
>> PicklingError: Can't pickle ui8[N]: it's not found as etils.enp.array_types.typing.ui8
```

Additionally, if `dill` is not available, the behavior doesn't change.

PiperOrigin-RevId: 642193800
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant