From 5a5d8a7affb37c606f159619ec24e64765974aae Mon Sep 17 00:00:00 2001
From: ebuehrle <43623224+ebuehrle@users.noreply.github.com>
Date: Fri, 25 Feb 2022 01:32:43 +0100
Subject: [PATCH] WIP: Support different plan lengths

---
 src/safe_options/options.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/safe_options/options.py b/src/safe_options/options.py
index e92128b..b4cfd8a 100644
--- a/src/safe_options/options.py
+++ b/src/safe_options/options.py
@@ -191,6 +191,7 @@ class SafeOptionsEnv(OptionsEnv):
             return np.ones(len(self.options), dtype=bool)
 
         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)
         safe = feasible(self.env, plans, method=self.safe_actions_collision_method)
         if not safe.any():