update to 0.6.2

This commit is contained in:
Balakumar Sundaralingam
2023-12-15 02:01:33 -08:00
parent d85ae41fba
commit 58958bbcce
105 changed files with 2514 additions and 934 deletions

View File

@@ -42,6 +42,7 @@ class WrapResult:
metrics: Optional[RolloutMetrics] = None
debug: Any = None
js_action: Optional[State] = None
raw_action: Optional[torch.Tensor] = None
def clone(self):
return WrapResult(
@@ -155,6 +156,7 @@ class WrapBase(WrapConfig):
solve_time=self.opt_dt,
metrics=metrics,
debug={"steps": self.get_debug_data(), "cost": self.get_debug_cost()},
raw_action=act_seq,
)
return result