google / google/orbax

Type Handler registry collision warnings

Open
#1,019 0 comments 1 reaction 1 assignee Claimed by @niketkumar View on GitHub
checkpoint type:bug
Dominant language
Python
Stars
535
Forks
101
Avg merge
1d 13h
Merged PRs (30d)
23

Description

When importing the `orbax` checkpointer, our scripts throw warnings as below:
```
WARNING:absl:Type handler registry overriding type "" collision on scalar
WARNING:absl:Type handler registry overriding type "" collision on scalar
WARNING:absl:Type handler registry overriding type "" 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:
```python
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.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.