Only remove time from parameters if present.

csu-hmc · May 11, 2024 · 006bc88 · 006bc88
1 parent 8d07245
commit 006bc88
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/opty/direct_collocation.py b/opty/direct_collocation.py
@@ -695,7 +695,8 @@ def _sort_parameters(self):
         The unknown parameters are sorted by name."""
 
         parameters = self.eom.free_symbols.copy()
-        parameters.remove(self.time_symbol)
+        if self.time_symbol in parameters:
+            parameters.remove(self.time_symbol)
 
         res = self._parse_inputs(parameters,
                                  self.known_parameter_map.keys())