admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 03:59:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adding flag for file writting recording case

Browse Source
This commit is contained in:
CarolinePascal
2025-04-09 15:00:18 +02:00
parent 52c424c5eb
commit 580008663b
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/microphones/microphone.py
+5 -1
View File
@@ -149,6 +149,7 @@ class Microphone:
self.logs = {} self.logs = {}
self.is_connected = False self.is_connected = False
self.is_recording = False self.is_recording = False
self.is_writing = False
def connect(self) -> None: def connect(self) -> None:
if self.is_connected: if self.is_connected:
@@ -208,7 +209,8 @@ class Microphone:
logging.warning(status) logging.warning(status)
# Slicing makes copy unnecessary # Slicing makes copy unnecessary
# Two separate queues are necessary because .get() also pops the data from the queue # Two separate queues are necessary because .get() also pops the data from the queue
self.record_queue.put(indata[:, self.channels]) if self.is_writing:
self.record_queue.put(indata[:, self.channels])
self.read_queue.put(indata[:, self.channels]) self.read_queue.put(indata[:, self.channels])
@staticmethod @staticmethod
@@ -312,6 +314,8 @@ class Microphone:
self.record_thread.daemon = True self.record_thread.daemon = True
self.record_thread.start() self.record_thread.start()
self.is_writing = True
self.is_recording = True self.is_recording = True
self.stream.start() self.stream.start()