Error Importing Ciclo

[Impure-King]

Hey, just wanted to test the library. However, it seems that it isn't being maintained regularly anymore. I got this error, when importing it:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
Cell In[1], line 4
      2 import sentinex as sx
      3 from sentinex import nn
----> 4 import ciclo as lo
      5 sx.__version__

File ~/miniconda3/envs/JAX/lib/python3.11/site-packages/ciclo/__init__.py:4
      1 __version__ = "0.1.9"
----> 4 from . import managed
      5 from .callbacks import checkpoint as checkpoint
      6 from .callbacks import early_stopping as early_stopping

File ~/miniconda3/envs/JAX/lib/python3.11/site-packages/ciclo/managed.py:30
     22 from ciclo.logging import Logs
     23 from ciclo.loops.loop import (
     24     CallbackOutput,
     25     FunctionCallbackOutputs,
   (...)
     28     to_standard_outputs,
     29 )
---> 30 from ciclo.strategies import Strategy, get_strategy
     31 from ciclo.types import Batch, Broadcasts, CluMetric, MetricLike, Statics
     32 from ciclo.utils import inject

File ~/miniconda3/envs/JAX/lib/python3.11/site-packages/ciclo/strategies.py:20
     15 class Dataclass(Protocol):
     16     __dataclass_fields__: Dict
     19 @runtime_checkable
---> 20 class HasKey(Protocol):
     21     key: jax.random.KeyArray
     24 StrategyConstructor = Callable[[], "Strategy"]

File ~/miniconda3/envs/JAX/lib/python3.11/site-packages/ciclo/strategies.py:21, in HasKey()
     19 @runtime_checkable
     20 class HasKey(Protocol):
---> 21     key: jax.random.KeyArray

File ~/miniconda3/envs/JAX/lib/python3.11/site-packages/jax/_src/deprecations.py:54, in deprecation_getattr.<locals>.getattr(name)
     52   warnings.warn(message, DeprecationWarning, stacklevel=2)
     53   return fn
---> 54 raise AttributeError(f"module {module!r} has no attribute {name!r}")

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