diff --git a/lavague-sdk/lavague/sdk/trajectory/model.py b/lavague-sdk/lavague/sdk/trajectory/model.py index d8157ff6..9e955fcd 100644 --- a/lavague-sdk/lavague/sdk/trajectory/model.py +++ b/lavague-sdk/lavague/sdk/trajectory/model.py @@ -101,10 +101,13 @@ def deserialize_action(cls, values: Dict[str, Any]) -> Dict[str, Any]: return values +class StepKnowledge(ActionWrapper): + instruction: Instruction + + class StepCompletion(ActionWrapper): run_status: RunStatus run_mode: RunMode - -class StepKnowledge(ActionWrapper): - instruction: Instruction + def to_knowledge(self, instruction: Instruction) -> StepKnowledge: + return StepKnowledge(instruction=instruction, action=self.action) \ No newline at end of file