Move lock to setter

damiafuentes · May 10, 2023 · c10bcc3 · c10bcc3
1 parent c5b55e3
commit c10bcc3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/djitellopy/tello.py b/djitellopy/tello.py
@@ -1064,8 +1064,7 @@ def update_frame(self):
                 if self.with_queue:
                     self.frames.append(np.array(frame.to_image()))
                 else:
-                    with self.lock:
-                        self.frame = np.array(frame.to_image())
+                    self.frame = np.array(frame.to_image())
 
                 if self.stopped:
                     self.container.close()
@@ -1096,7 +1095,8 @@ def frame(self):
 
     @frame.setter
     def frame(self, value):
-        self._frame = value
+        with self.lock:
+            self._frame = value
 
     def stop(self):
         """Stop the frame update worker