ruff format

This commit is contained in:
Abinila Siva 2025-11-24 11:51:17 -05:00
parent 4570808030
commit 14f7d27274
2 changed files with 8 additions and 15 deletions

View File

@ -372,10 +372,7 @@ class MemryXDetector(DetectionApi):
return None return None
try: try:
# Wait for a frame from the queue with timeout to check stop_event periodically # Wait for a frame from the queue with timeout to check stop_event periodically
frame = self.capture_queue.get( frame = self.capture_queue.get(block=True, timeout=0.5)
block=True,
timeout=0.5
)
return frame return frame
@ -390,11 +387,8 @@ class MemryXDetector(DetectionApi):
"""Retrieve processed results from MemryX output queue + a copy of the original frame""" """Retrieve processed results from MemryX output queue + a copy of the original frame"""
try: try:
# Get connection ID with timeout # Get connection ID with timeout
connection_id = ( connection_id = self.capture_id_queue.get(
self.capture_id_queue.get( block=True, timeout=1.0
block=True,
timeout=1.0
)
) # Get the corresponding connection ID ) # Get the corresponding connection ID
detections = self.output_queue.get() # Get detections from MemryX detections = self.output_queue.get() # Get detections from MemryX
@ -410,7 +404,6 @@ class MemryXDetector(DetectionApi):
return None, None return None, None
def post_process_yolonas(self, output): def post_process_yolonas(self, output):
predictions = output[0] predictions = output[0]