huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix typo

Browse Source
This commit is contained in:
Johannes Fischer
2021-09-10 11:34:13 +02:00
parent 5ff4b42c0e
commit b2b2abafa2
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scratch/etienne/intersimple/gail_options.py
View File

@@ -47,11 +47,12 @@ def target_velocity_plan(current_v: float, target_v: float, t: int)
# for now, constant acceleration # for now, constant acceleration
a = (target_v - current_v) / t a = (target_v - current_v) / t
return a*np.ones((t,)) return a*np.ones((t,))
Hello
def generate_plan(env, i): def generate_plan(env, i):
"""Generate input profile for high-level action `i`.""" """Generate input profile for high-level action `i`."""
assert i < len(ALL_OPTIONS), "Invalid option index {i}" assert i < len(ALL_OPTIONS), "Invalid option index {i}"
target_v, t = ALL_OPTSIONS[i] target_v, t = ALL_OPTIONS[i]
current_v = env._env.state[env._agent, 1].item() # extract from env current_v = env._env.state[env._agent, 1].item() # extract from env
plan = target_velocity_plan(current_v, target_v, t) plan = target_velocity_plan(current_v, target_v, t)
assert len(plan) == t, "incorrect plan length" assert len(plan) == t, "incorrect plan length"