remove comment

zaxtax · zaxtax · commit 93a51c79654a · 2022-12-06T14:50:48.000+01:00
diff --git a/flax/serialization.py b/flax/serialization.py
@@ -193,8 +193,6 @@ def _restore_namedtuple(xs, state_dict: Dict[str, Any]):
                              _namedtuple_state_dict,
                              _restore_namedtuple)
 
-# jax.tree_util.Partial does not support flax serialization
-# should be upstreamed to Flax
 register_serialization_state(
     jax.tree_util.Partial,
     lambda x: (
