mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-15 08:39:49 +00:00
Adding flag for file writting recording case
This commit is contained in:
@@ -149,6 +149,7 @@ class Microphone:
|
||||
self.logs = {}
|
||||
self.is_connected = False
|
||||
self.is_recording = False
|
||||
self.is_writing = False
|
||||
|
||||
def connect(self) -> None:
|
||||
if self.is_connected:
|
||||
@@ -208,7 +209,8 @@ class Microphone:
|
||||
logging.warning(status)
|
||||
# Slicing makes copy unnecessary
|
||||
# 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])
|
||||
|
||||
@staticmethod
|
||||
@@ -312,6 +314,8 @@ class Microphone:
|
||||
self.record_thread.daemon = True
|
||||
self.record_thread.start()
|
||||
|
||||
self.is_writing = True
|
||||
|
||||
self.is_recording = True
|
||||
self.stream.start()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user