Eclectic-Sheep / sheeprl

Distributed Reinforcement Learning accelerated by Lightning Fabric
https://eclecticsheep.ai
Apache License 2.0
275 stars 26 forks source link

Import user-defined evaluation file #199

Closed belerico closed 5 months ago

belerico commented 5 months ago

Summary

In the sheeprl.cli.eval_algorithm method we still import the evaluation file with task = importlib.import_module(f"{module}.evaluate"), with the sheeprl/{algo_name}/evaluate.py file that could be missing since the user could have changed its name. The entrypoint filename can be retrieved from the evaluation_registry.

Type of Change

Please select the one relevant option below:

Checklist

Please confirm that the following tasks have been completed:

Thank you for your contribution! Once you have filled out this template, please ensure that you have assigned the appropriate reviewers and that all tests have passed.