Skip to content

Pin dependencies for RLlib examples to avoid incompatibilities#315

Open
Sujit-Hiwale wants to merge 1 commit intogoogle-deepmind:mainfrom
Sujit-Hiwale:fix-rllib-examples-requirements
Open

Pin dependencies for RLlib examples to avoid incompatibilities#315
Sujit-Hiwale wants to merge 1 commit intogoogle-deepmind:mainfrom
Sujit-Hiwale:fix-rllib-examples-requirements

Conversation

@Sujit-Hiwale
Copy link

Summary

This PR updates examples/requirements.in to pin dependency versions that are
known to work with the RLlib example scripts.

Without these pins, installing the example dependencies on modern Python
environments often results in runtime errors due to incompatibilities between
Ray 2.5, NumPy 2.x, SciPy, TensorFlow, JAX, and related libraries.

The updated requirements reflect a tested, working configuration and improve
reproducibility for users running the RLlib examples.


Motivation

Several users (see #299) have reported failures when installing and running the
example training scripts using the current examples/requirements.in. The file
does not specify version constraints for a number of critical dependencies,
leading to incompatible combinations being installed by default.

This change makes the dependency expectations explicit and aligns the
requirements with versions that successfully run the example training code.


Changes

  • Pin NumPy, SciPy, Ray, TensorFlow, TensorFlow Probability, JAX, Chex, and
    typing_extensions to compatible versions
  • Add TensorFlow dependencies required for loading pretrained Melting Pot bots
  • Keep the scope limited to example dependencies only (no core code changes)

Testing

  • Installed dependencies from examples/requirements.in in a clean virtual
    environment
  • Verified core imports (Ray, TensorFlow, JAX, Melting Pot)
  • Successfully launched the RLlib self-play example
    (python -m examples.rllib.self_play_train) without runtime errors

This change does not affect the core Melting Pot library and only impacts the
example setup.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant