Possible Bugs in CQL_SAC Example

[ChengDaHaI]

Hi,

I am running the cql example (agents/tf_agents/examples/cql_sac/kumar20/cql_sac_train_eval.py) with tf-agents v0.15.0 + tf v2.11.0.

It raises error

AttributeError: module 'rlds' has no attribute 'load'

from the line

    load_dataset_fn: Optional[Callable[[str], tf.data.Dataset]] = rlds.load,

• It should be a bug and it could be fixed by changing rlds.load to tfds.load.

• If it is not a bug, how to solve the error?