Skip to content

Commit

Permalink
test github compatibility
Browse files Browse the repository at this point in the history
  • Loading branch information
AlirezaMorsali committed Oct 7, 2022
1 parent 2c454bb commit 4600214
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -94,7 +94,7 @@ response = iai.initialize(
> _response["states"]_ is a list of agent states, by default the first on the list is always the ego vehicle.
## DRIVE
**DRIVE** is Inverted AI's cutting-edge realistic driving model trained on millions of miles of traffic data.
{py:func}`invertedai.api_resources.drive` **DRIVE** is Inverted AI's cutting-edge realistic driving model trained on millions of miles of traffic data.
This model can drive all the agents with only the current state of the environment, i.e., one step observations (which could be obtained from **INITIALIZE**) or with multiple past observations.
```python
response = iai.drive(
Expand Down
1 change: 0 additions & 1 deletion docs/source/sdk.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,5 +4,4 @@
.. autofunction:: invertedai.api_resources.drive
.. autofunction:: invertedai.api_resources.available_maps
.. autofunction:: invertedai.api_resources.get_map
```
3 changes: 2 additions & 1 deletion invertedai/api_resources.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,17 +83,18 @@ def initialize(


def drive(
location="CARLA:Town03:Roundabout",
states: dict,
agent_attributes: dict,
recurrent_states: Optional[List] = None,
get_birdviews: bool = False,
location="CARLA:Town03:Roundabout",
steps: int = 1,
get_infractions: bool = False,
traffic_states_id: str = "000:0",
exclude_ego_agent: bool = True,
present_mask: Optional[List] = None
) -> dict:

def _tolist(input_data: List):
if not isinstance(input_data, list):
return input_data.tolist()
Expand Down

0 comments on commit 4600214

Please sign in to comment.