orbax icon indicating copy to clipboard operation
orbax copied to clipboard

Type Handler registry collision warnings

Open varunagrawal opened this issue 6 months ago • 0 comments

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.

varunagrawal avatar Jul 25 '24 18:07 varunagrawal