code-loader 0.2.71.dev11__py3-none-any.whl → 0.2.71.dev12__py3-none-any.whl

code_loader/samples_generator_parallelized.py

@@ -79,25 +79,25 @@ class SamplesGeneratorParallelized:
         self.start()
         return self._generate_samples(sample_identities)
 
+    @staticmethod
+    def _release_queue(queue: Queue):
+        assert queue is not None
+        while not queue.empty():
+            queue.get()
+        queue.close()
+        queue.join_thread()
+
     def release(self) -> None:
         if self.processes is None:
             return
         for process in self.processes:
             process.terminate()
+            process.kill()
             process.join()
             process.close()
 
-        assert self._samples_to_process is not None
-        while not self._samples_to_process.empty():
-            self._samples_to_process.get()
-        self._samples_to_process.close()
-        self._samples_to_process.join_thread()
-
-        assert self._ready_samples is not None
-        while not self._ready_samples.empty():
-            self._ready_samples.get()
-        self._ready_samples.close()
-        self._ready_samples.join_thread()
+        self._release_queue(self._samples_to_process)
+        self._release_queue(self._ready_samples)
 
         self.processes = None
 

{code_loader-0.2.71.dev11.dist-info → code_loader-0.2.71.dev12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 0.2.71.dev11
+Version: 0.2.71.dev12
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-0.2.71.dev11.dist-info → code_loader-0.2.71.dev12.dist-info}/RECORD

@@ -20,11 +20,11 @@ code_loader/leap_binder/leapbinder.py,sha256=WjC-UA5xeY9UZtMqJQOxjidlFcWURC9mmy7
 code_loader/leaploader.py,sha256=3vvzBzS0oev6kqXjtriz3G6z0tijGOJLIH8QoOvu1Ls,15044
 code_loader/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/metrics/default_metrics.py,sha256=xtvTskR7LC5pyKSlFcvlyBpFY9TSXNGVAi5taEPSI60,6727
-code_loader/samples_generator_parallelized.py,sha256=I9NGFTEMMDTEjfazAtbygrSNceE3MIeTG0n0h2AF4O4,3635
+code_loader/samples_generator_parallelized.py,sha256=f822mhaJUXNATOu5NvIUuNR4WMvUS8hrPbjjMharHgg,3516
 code_loader/utils.py,sha256=WUcM97OuxrhfLCRPoH9EbXrxajNpYgX1CTMc3_PXtYU,1736
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=HqWx2qfTrroGl2n8Fpmr_4X-rk7tE2oGapjO3gzz4WY,2226
-code_loader-0.2.71.dev11.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-0.2.71.dev11.dist-info/METADATA,sha256=x_XungMYF_G0OBKQqnqiHBCC0B016l9L-dOCDwNmNr4,882
-code_loader-0.2.71.dev11.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-code_loader-0.2.71.dev11.dist-info/RECORD,,
+code_loader-0.2.71.dev12.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-0.2.71.dev12.dist-info/METADATA,sha256=D1vskJ1XvWKd1GhFfPB80UzH7OUuCkd5qPfd4d6XkOQ,882
+code_loader-0.2.71.dev12.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+code_loader-0.2.71.dev12.dist-info/RECORD,,