Add __dealloc__() method to release resources

commaai · Dec 18, 2024 · 830c17b · 830c17b
1 parent ff97a43
commit 830c17b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/selfdrive/modeld/models/commonmodel_pyx.pyx b/selfdrive/modeld/models/commonmodel_pyx.pyx
@@ -67,10 +67,16 @@ cdef class DrivingModelFrame(ModelFrame):
     self.frame = <cppModelFrame*>(self._frame)
     self.buf_size = self._frame.buf_size
 
+  def __dealloc__(self):
+    del self._frame
+
 cdef class MonitoringModelFrame(ModelFrame):
   cdef cppMonitoringModelFrame * _frame
 
   def __cinit__(self, CLContext context):
     self._frame = new cppMonitoringModelFrame(context.device_id, context.context)
     self.frame = <cppModelFrame*>(self._frame)
     self.buf_size = self._frame.buf_size
+
+  def __dealloc__(self):
+    del self._frame