Skip to content

Commit

Permalink
Add NeurIPS 2022 Control and Supervised Learning agent to agent zoo (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
Adaickalavan authored Feb 12, 2023
1 parent 4d57e09 commit a81c3ce
Show file tree
Hide file tree
Showing 5 changed files with 48 additions and 2 deletions.
3 changes: 2 additions & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ Copy and pasting the git commit messages is __NOT__ enough.

## [Unreleased]
### Added
- Added a zoo agent, named Control-and-Supervised-Learning, from NeurIPS 2022 submission. This zoo agent runs in benchmark `driving_smarts==0.0`.
### Changed
### Deprecated
### Fixed
Expand All @@ -21,7 +22,7 @@ Copy and pasting the git commit messages is __NOT__ enough.
- Engine configuration utility that uses the following locations to allow configuration of the SMARTS engine. The engine consumes the configuration files from the following locations in the following priority: `./engine.ini`, `~/.smarts/engine.ini`, `$GLOBAL_USER/smarts/engine.ini`, and `${PYTHON_ENV}/lib/${PYTHON_VERSION}/site-packages/smarts/engine.ini`.
- Added map source uri as `map_source` inside of `hiway-v1` reset info to indicate what the current map is on reset.
- Added NGSIM documentation.
- Added a zoo agent, named Interaction-aware Motion Prediction, from NeurIPS2022 submission. This zoo agent runs in benchmark `driving_smarts==0.0`.
- Added a zoo agent, named Interaction-Aware Motion Prediction, from NeurIPS2022 submission. This zoo agent runs in benchmark `driving_smarts==0.0`.
- Added Agent Zoo documentation in ReadTheDocs.
### Changed
- Made changes in the docs to reflect `master` branch as the main development branch.
Expand Down
8 changes: 7 additions & 1 deletion docs/benchmarks/agent_zoo.rst
Original file line number Diff line number Diff line change
Expand Up @@ -53,4 +53,10 @@ Available zoo agents
- driving_smarts==0.0
- :attr:`~smarts.core.controllers.ActionSpaceType.TargetPose`
- `code <https://github.com/smarts-project/smarts-project.rl/tree/master/interaction_aware_motion_prediction>`__
- Contributed as part of `NeurIPS 2022 Driving SMARTS <https://smarts-project.github.io/archive/2022_nips_driving_smarts/>`_ competition.
- Contributed as part of `NeurIPS 2022 Driving SMARTS <https://smarts-project.github.io/archive/2022_nips_driving_smarts/>`__ competition.
* - | zoo.policies:control-and-supervised-learning-agent-v0
| zoo/policies/control_and_supervised_learning
- driving_smarts==0.0
- :attr:`~smarts.core.controllers.ActionSpaceType.TargetPose`
- `code <https://github.com/smarts-project/smarts-project.rl/tree/master/control_and_supervised_learning>`__
- Contributed as part of `NeurIPS 2022 Driving SMARTS <https://smarts-project.github.io/archive/2022_nips_driving_smarts/>`__ competition.
16 changes: 16 additions & 0 deletions zoo/policies/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,3 +133,19 @@ def entry_point_iamp(**kwargs):
register(
locator="interaction-aware-motion-prediction-agent-v0", entry_point=entry_point_iamp
)

def entry_point_casl(**kwargs):
pkg = "control_and_supervised_learning"
module = ".policy"
lib = _verify_installation(pkg=pkg, module=module)

return AgentSpec(
interface=AgentInterface(
action=ActionSpaceType.TargetPose,
),
agent_builder=lib.Policy,
)

register(
locator="control-and-supervised-learning-agent-v0", entry_point=entry_point_casl,
)
19 changes: 19 additions & 0 deletions zoo/policies/control_and_supervised_learning/setup.cfg
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
[metadata]
name = smarts_agent_casl_stub
version = 0.1.0
url = https://github.com/huawei-noah/SMARTS
description = SMARTS zoo agent.
long_description = Trained Control and Supervised Learning agent. See [SMARTS](https://github.com/huawei-noah/SMARTS).
long_description_content_type=text/markdown
classifiers=
Programming Language :: Python
Programming Language :: Python :: 3 :: Only
Programming Language :: Python :: 3.8

[options]
packages = find:
include_package_data = True
zip_safe = True
python_requires = == 3.8.*
install_requires =
smarts_agent_casl==0.1.0
4 changes: 4 additions & 0 deletions zoo/policies/control_and_supervised_learning/setup.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
from setuptools import setup

if __name__ == "__main__":
setup()

0 comments on commit a81c3ce

Please sign in to comment.