wilson-labs/cola

[Bug] Deprecated PyTree Function in Pytorch 2.2

adam-hartshorne opened this issue · 0 comments

The latest version of PyTorch has deprecated torch.utils._pytree._register_pytree_node in favour of torch.utils._pytree.register_pytree_node, which results in many of the following warnings,

...site-packages/cola/backends/backends.py:75: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.