ray_utilities package
Subpackages
- ray_utilities.callbacks package
- Subpackages
- ray_utilities.callbacks.algorithm package
- Submodules
- ray_utilities.callbacks.algorithm.adjust_config_based_on_batch_size module
- ray_utilities.callbacks.algorithm.algo_metric_checkpointer module
- ray_utilities.callbacks.algorithm.callback_mixins module
- ray_utilities.callbacks.algorithm.discrete_eval_callback module
- ray_utilities.callbacks.algorithm.dynamic_batch_size module
- ray_utilities.callbacks.algorithm.dynamic_buffer_callback module
- ray_utilities.callbacks.algorithm.dynamic_evaluation_callback module
- ray_utilities.callbacks.algorithm.dynamic_hyperparameter module
- ray_utilities.callbacks.algorithm.env_render_callback module
- ray_utilities.callbacks.algorithm.exact_sampling_callback module
- ray_utilities.callbacks.algorithm.model_config_saver_callback module
- ray_utilities.callbacks.algorithm.reset_episode_metrics module
- ray_utilities.callbacks.algorithm.seeded_env_callback module
- Module contents
- ray_utilities.callbacks.tuner package
- Submodules
- ray_utilities.callbacks.tuner.adv_comet_callback module
- ray_utilities.callbacks.tuner.adv_csv_callback module
- ray_utilities.callbacks.tuner.adv_json_logger_callback module
- ray_utilities.callbacks.tuner.adv_tbx_logger_callback module
- ray_utilities.callbacks.tuner.adv_wandb_callback module
- ray_utilities.callbacks.tuner.metric_checkpointer module
- ray_utilities.callbacks.tuner.new_style_logger_callback module
- Module contents
- ray_utilities.callbacks.algorithm package
- Submodules
- ray_utilities.callbacks.progress_bar module
- Module contents
- Subpackages
- ray_utilities.config package
- ray_utilities.connectors package
- ray_utilities.dynamic_config package
- ray_utilities.environment package
- ray_utilities.jax package
- Subpackages
- Submodules
- ray_utilities.jax.jax_learner module
- ray_utilities.jax.jax_model module
- ray_utilities.jax.jax_module module
- ray_utilities.jax.math module
- ray_utilities.jax.utils module
- Module contents
- ray_utilities.learners package
- ray_utilities.runfiles package
- ray_utilities.setup package
- Submodules
- ray_utilities.setup.algorithm_setup module
- ray_utilities.setup.experiment_base module
- ray_utilities.setup.extensions module
- ray_utilities.setup.optuna_setup module
- ray_utilities.setup.ppo_mlp_setup module
- ray_utilities.setup.scheduled_tuner_setup module
- ray_utilities.setup.tuner_setup module
- Module contents
- ray_utilities.training package
- ray_utilities.tune package
- ray_utilities.typing package
- Submodules
- ray_utilities.typing.algorithm_return module
- ray_utilities.typing.common module
- ray_utilities.typing.discrete_module module
- ray_utilities.typing.generic_rl_module module
- ray_utilities.typing.jax module
- ray_utilities.typing.metrics module
- ray_utilities.typing.model_return module
- ray_utilities.typing.trainable_return module
- Module contents
- ray_utilities.video package
Submodules
ray_utilities.comet module
ray_utilities.constants module
Constants and configuration values used throughout Ray Utilities.
Defines important constants, version checks, and configuration values used across the Ray Utilities package. Includes version compatibility flags, metric keys for Ray RLlib, Comet ML configuration, and video logging constants.
- Key Constants:
RAY_VERSION: Current Ray version for compatibility checksRAY_NEW_API_STACK_ENABLED: Whether Ray’s new API stack is availableEVAL_METRIC_RETURN_MEAN: Standard evaluation return metric keyDEFAULT_VIDEO_DICT_KEYS: Video logging configuration keysCOMET_OFFLINE_DIRECTORY: Path for offline Comet ML experiments
Example
>>> from ray_utilities.constants import RAY_NEW_API_STACK_ENABLED
>>> if RAY_NEW_API_STACK_ENABLED:
... # Use new Ray API features
... pass