orbax
orbax copied to clipboard
Type Handler registry collision warnings
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.