Skip to content

Type Handler registry collision warnings #1019

Open
@varunagrawal

Description

@varunagrawal

When importing the orbax checkpointer, our scripts throw warnings as below:

WARNING:absl:Type handler registry overriding type "<class 'float'>" collision on scalar
WARNING:absl:Type handler registry overriding type "<class 'bytes'>" collision on scalar
WARNING:absl:Type handler registry overriding type "<class 'numpy.number'>" collision on scalar

Since these are warnings I don't think there is anything particularly bad that needs remedying but we would like to remove the warnings for our logging.

A MRE is given below:

from orbax import checkpoint as ocp

print("Testing the OCP warnings")

Is this an issue with how the global type registry is set up? Thanks in advance.

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions