From 104bdd1d506f80a932b303a435222ae6c5300bc3 Mon Sep 17 00:00:00 2001
From: Sgr A* VMT <1611902585@qq.com>
Date: Tue, 11 Jun 2024 08:39:36 +0000
Subject: [PATCH] fix

Signed-off-by: Sgr A* VMT <1611902585@qq.com>
---
 idm.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/idm.py b/idm.py
index 90490a1..8931f15 100644
--- a/idm.py
+++ b/idm.py
@@ -1275,7 +1275,14 @@ class IDMProbeWrapper:
         return self.idm.get_lift_speed(gcmd)
     def run_probe(self, gcmd):
         return self.idm.run_probe(gcmd)
-
+    def get_probe_params(self, gcmd=None):
+        return {'probe_speed': self.idm.speed,
+            'lift_speed': self.idm.lift_speed}
+    def start_probe_session(self, gcmd):
+        self.multi_probe_begin()
+        return self
+    def end_probe_session(self):
+        self.multi_probe_end()
 class IDMTempWrapper:
     def __init__(self, idm):
         self.idm = idm