Feature/add build agents - Githubissues

Eclectic-Sheep / sheeprl

Distributed Reinforcement Learning accelerated by Lightning Fabric

https://eclecticsheep.ai

Apache License 2.0

282 stars 29 forks source link

Feature/add build agents #153

Closed michele-milesi closed 8 months ago

michele-milesi commented 8 months ago

Summary

Describe the purpose of the pull request, including:

Renamed the build_models() function to build_agent() for all the Dreamer-like algorithms.
Added build_agent() function to all the algorithms.

Type of Change

Please select the one relevant option below:

New feature (non-breaking change that adds functionality)

Checklist

Please confirm that the following tasks have been completed:

[x] I have tested my changes locally and they work as expected. (Please describe the tests you performed.)
[x] I have added unit tests for my changes, or updated existing tests if necessary.
[x] I have updated the documentation, if applicable.
[x] I have installed pre-commit and run locally for my code changes.