Skip to main content

Environments for reinforcement learning

Project description

rl-envs-forge

GitHub PyPI version License: MIT Code style: black

Lightweight environments for reinforcement learning applications.

Table of Contents

Installation

pip install rl-envs-forge

Environments

Labyrinth

Labyrinth is a classic maze-solving environment, where the goal is to navigate from a start point to a target. The maze layout is randomly generated based on a set of parametrizable arguments.

📖 Detailed Documentation: Click here to read more about the Labyrinth environment

Labyrinth rendered example

Labyrinth render GIF

KArmedBandit

KArmedBandit is a bandit environment, which returns a reward from a distribution associated with the chosen arm at each timestep. This implementation includes multiple distributions, and the possibility to shift the distribution parameters during sampling.

📖 Detailed Documentation: Click here to read more about the KArmedBandit environment

KArmedBandit rendered example

KArmedBandit render

GridWorld

GridWorld is a customizable grid-based environment for reinforcement learning, featuring adjustable grid size, start and terminal states, walls, and special transitions. Each action taken by the agent results in a transition within the grid, adhering to the defined rules and probabilities.

📖 Detailed Documentation: Click here to read more about the GridWorld environment

GridWorld rendered example

GridWorld render

ACML

The Adaptive Computation and Machine Learning (ACML) environments are toy environments proposed in Reinforcement Learning: An Introduction* (2nd ed.).

📖 Detailed Documentation: Click here to read more about the ACML environments

Inverted pendulum environments

Inverted pendulum environments where the objective is to apply forces to maintain the pendulum upright despite disturbances and the natural tendency to fall.

📖 Detailed Documentation: Click here to read more about the inverted pendulum envs

Inverted pendulums rendered examples

CartPole PendulumDisk
CartPole PendulumDisk

NetworkGraph

NetworkGraph is an environment simulating the current opinion in a social network.

📖 Detailed Documentation: Click here to read more about the NetworkGraph environment

NetworkGraph rendered example

NetworkGraph render

Usage

Example code on setting up and testing the Labyrinth environment.

Note, this code snippet produced the render visible in section Labyrinth

from time import sleep
from rl_envs_forge.envs.labyrinth.labyrinth import Labyrinth

env = Labyrinth(20, 20, seed=0)

done = False
quit_event = False
while not done and not quit_event:
    action = env.action_space.sample()  
    observation, reward, done, truncated, info = env.step(action)
    quit_event, _ = env.render()
    sleep(0.1)

Tests

Requirements: pytest and pytest-cov

Run the tests in the root folder with:

pytest tests

License

This project is licensed under the MIT License.

Contact & Support

For any queries or support, or if you would like to contribute to this project, reach out at marius.dragomir.dgm@gmail.com or raise an issue on our GitHub repository.

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

rl_envs_forge-5.22.0.tar.gz (69.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

rl_envs_forge-5.22.0-py3-none-any.whl (88.2 kB view details)

Uploaded Python 3

File details

Details for the file rl_envs_forge-5.22.0.tar.gz.

File metadata

  • Download URL: rl_envs_forge-5.22.0.tar.gz
  • Upload date:
  • Size: 69.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.3.4 CPython/3.14.3 Linux/6.17.0-1010-azure

File hashes

Hashes for rl_envs_forge-5.22.0.tar.gz
Algorithm Hash digest
SHA256 c562205c5dcf8b4c0a388fcacb245aa69ec4b8fa1d97689e367fc338d963201f
MD5 c6f5cdbe242d8c997f2935df5f180b62
BLAKE2b-256 953104e66b41d16df10c3aa3f9be6a7b2c713101ef39a8c72ac556f6e6232eb1

See more details on using hashes here.

File details

Details for the file rl_envs_forge-5.22.0-py3-none-any.whl.

File metadata

  • Download URL: rl_envs_forge-5.22.0-py3-none-any.whl
  • Upload date:
  • Size: 88.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.3.4 CPython/3.14.3 Linux/6.17.0-1010-azure

File hashes

Hashes for rl_envs_forge-5.22.0-py3-none-any.whl
Algorithm Hash digest
SHA256 bf19cff3421bb69deaf5a3458c3d4a53dfa43288f10fcac779b4b4252b90b096
MD5 8a48e837252e9fcf93c4c41e5d2959b8
BLAKE2b-256 06d1b8b051759c3f573167d81ac0a00144782bde6f56dcd625ebf1c455863f76

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page