ray
ray copied to clipboard
[RLlib] Fix invalid call of action_sampler_fn to support Keras 3
Fixes: https://github.com/ray-project/ray/issues/44671 @sven1977 could you take a look?
Should be merged together with https://github.com/ray-project/ray/pull/45562 to improve Keras 3 support.
@ArturNiederfahrenhorst could you take a look? I would appreciate it! Would love to see Keras 3 support in rllib, and it is just two small PRs away ;)