KerasでCallbackListがインポートできない

Question

強化学習のDQN(DDQN)で多腕バンディット問題を実装しようとしています。
そこで、まず次のサイトを参考にしました。
https://qiita.com/pocokhc/items/fd133053fa309bdb58e6

そこのリンクで次のコードを実行してみました。
https://gist.github.com/pocokhc/25ab6f208a940094c7429d1e8b06ea57

そうしましたら、次のエラーが出ました。

Traceback (most recent call last):
  File "qiita_03_dqn.py", line 15, in <module>
    import rl.core
  File "/home/********/.local/lib/python3.6/site-packages/rl/core.py", line 8, in <module>
    from rl.callbacks import (
  File "/home/********/.local/lib/python3.6/site-packages/rl/callbacks.py", line 11, in <module>
    from keras.callbacks import Callback as KerasCallback, CallbackList as KerasCallbackList
ImportError: cannot import name 'CallbackList'

エラーを注意深く読んでみたところ、これは「keras.callbacksからCallbackListがインポートできないよ」というエラーのようです。
しかし、kerasは最新のバージョンですし、kerasにはCallbackListというメソッドがあるようです。
この問題の解決法がおわかりの方よろしくお願いします。

環境は次のとおりです。
OS: ubuntu3.8
Python: 3.8
tensorflow: 2.2.0
keras: 2.4.3

Answer 1

import文を以下のようにすると直るかもしれません。

from tensorflow.python.keras.callbacks import CallbackList as KerasCallbackList