OWL-ViT inference playround: [scenic.projects.owl_vit import error]

[tony-octo-ibm]

AttributeError Traceback (most recent call last)
/tmp/ipython-input-572887953.py in <cell line: 0>()
8 import numpy as np
9 from scenic.projects.owl_vit import configs
---> 10 from scenic.projects.owl_vit import models
11
12 from scenic.projects.owl_vit.notebooks import inference

8 frames
/usr/local/lib/python3.12/dist-packages/ott/initializers/nn/initializers.py in MetaInitializer()
62 meta_model: Optional[nn.Module] = None,
63 opt: optax.GradientTransformation = optax.adam(learning_rate=1e-3),
---> 64 rng: jax.random.PRNGKeyArray = jax.random.PRNGKey(0),
65 state: Optional[train_state.TrainState] = None
66 ):

AttributeError: module 'jax.random' has no attribute 'PRNGKeyArray'