with orbax checkpoint version '0.5.3'

with ocp.CheckpointManager(ckpt_dir, options=check_options, item_names=('state', 'metadata')) as mngr:
  mngr.save(...)

I got

TypeError: 'CheckpointManager' object does not support the context manager protocol

So I wander if your solution is still valid or if it is for future version??