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

WIP: Support different plan lengths

Browse Source
This commit is contained in:
ebuehrle
2022-02-25 01:32:43 +01:00
parent 02c1813b00
commit 5a5d8a7aff
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/safe_options/options.py
View File

@@ -191,6 +191,7 @@ class SafeOptionsEnv(OptionsEnv):
return np.ones(len(self.options), dtype=bool) return np.ones(len(self.options), dtype=bool)
plans = [self.plan(o) for o in self.options] plans = [self.plan(o) for o in self.options]
plans = [np.pad(p, (0, self.max_plan_length - len(p)), constant_values=np.nan) for p in plans]
plans = np.stack(plans) plans = np.stack(plans)
safe = feasible(self.env, plans, method=self.safe_actions_collision_method) safe = feasible(self.env, plans, method=self.safe_actions_collision_method)
if not safe.any(): if not safe.any():