@@ -42,15 +42,21 @@ def remove_shm_from_resource_tracker():
4242 def fix_register (name , rtype ):
4343 if rtype == "shared_memory" :
4444 return
45- return resource_tracker ._resource_tracker .register (
46- self , name , rtype )
45+ try :
46+ return resource_tracker ._resource_tracker .register (
47+ self , name , rtype )
48+ except NameError :
49+ return None
4750 resource_tracker .register = fix_register
4851
4952 def fix_unregister (name , rtype ):
5053 if rtype == "shared_memory" :
5154 return
52- return resource_tracker ._resource_tracker .unregister (
53- self , name , rtype )
55+ try :
56+ return resource_tracker ._resource_tracker .unregister (
57+ self , name , rtype )
58+ except NameError :
59+ return None
5460 resource_tracker .unregister = fix_unregister
5561
5662 if "shared_memory" in resource_tracker ._CLEANUP_FUNCS :
@@ -350,7 +356,7 @@ def _dequeue(self):
350356 return interactions
351357
352358 @abstractmethod
353- def enqueue (self ):
359+ def enqueue (self , data ):
354360 pass # pragma: no cover
355361
356362 @abstractmethod
@@ -618,7 +624,9 @@ def write_into(self, w, h, np_arr):
618624 self .info_buffer_repr [2 + next_buffer_index * 2 + 1 ] = h
619625 self .info_buffer_repr [1 ] = next_buffer_index
620626
621- def get_current_frame (self , from_buffer = False ):
627+ def get_current_frame (self ):
628+ """Get the current frame from the buffer.
629+ """
622630 if not self ._use_shared_mem :
623631 image_info = np .frombuffer (
624632 self .info_buffer , _UINT_ShM_TYPE )
@@ -636,6 +644,12 @@ def get_current_frame(self, from_buffer=False):
636644 return self .width , self .height , image
637645
638646 def get_jpeg (self ):
647+ """Returns a jpeg image from the buffer.
648+
649+ Returns:
650+ bytes: jpeg image.
651+
652+ """
639653 width , height , image = self .get_current_frame ()
640654
641655 if self ._use_shared_mem :
0 commit comments