code-loader 0.2.84.dev1__py3-none-any.whl → 0.2.84.dev3__py3-none-any.whl

code_loader/leap_loader_parallelized_base.py

@@ -1,12 +1,10 @@
 # mypy: ignore-errors
 import multiprocessing
 from abc import ABC, abstractmethod
-from dataclasses import dataclass
 from functools import lru_cache
 from queue import Empty
 from threading import Thread
-from typing import List, Tuple, Optional, Dict, Any
-import tensorflow as tf
+from typing import List, Optional, Any
 from multiprocessing import Process, Queue
 import psutil
 

code_loader/metric_calculator_parallelized.py

@@ -27,7 +27,7 @@ class MetricCalculatorParallelized(LeapLoaderParallelizedBase):
             target=MetricCalculatorParallelized._process_func,
             args=(self.code_path, self.code_entry_name, self._inputs_waiting_to_be_process,
                   self._ready_processed_results))
-        process.daemon = True
+        process.daemon = False
         process.start()
         return process
 
@@ -38,7 +38,8 @@ class MetricCalculatorParallelized(LeapLoaderParallelizedBase):
         while True:
             metric_id, metric_name, input_arg_name_to_tensor = samples_to_process.get(block=True)
             try:
-                metric_result = leap_loader.metric_by_name[metric_name].function(**input_arg_name_to_tensor)
+                with tf.device('/cpu:0'):
+                    metric_result = leap_loader.metric_by_name()[metric_name].function(**input_arg_name_to_tensor)
             except Exception as e:
                 leap_script_trace = traceback.format_exc().split('File "<string>"')[-1]
                 ready_samples.put(MetricSerializableError(metric_id, metric_name, leap_script_trace, str(e)))
@@ -46,6 +47,5 @@ class MetricCalculatorParallelized(LeapLoaderParallelizedBase):
 
             ready_samples.put((metric_id, metric_result))
 
-    def calculate_metrics(self, input_arg_name_to_tensor_list: List[Tuple[str, Dict[str, tf.Tensor]]]):
+    def calculate_metrics(self, input_arg_name_to_tensor_list: List[Tuple[str, str, Dict[str, tf.Tensor]]]):
         return self.start_process_inputs(input_arg_name_to_tensor_list)
-

{code_loader-0.2.84.dev1.dist-info → code_loader-0.2.84.dev3.dist-info}/METADATA

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

{code_loader-0.2.84.dev1.dist-info → code_loader-0.2.84.dev3.dist-info}/RECORD

@@ -20,16 +20,16 @@ code_loader/helpers/instancesegmentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
 code_loader/helpers/instancesegmentation/utils.py,sha256=bl4nlg7mV8CZyPl8grGAE5603TFVM7PTZyM3BRMbXBg,930
 code_loader/leap_binder/__init__.py,sha256=Oe7DNj6RmgX-tMruOrI7GYvf3SyAnzOxugF2e117_Z8,93
 code_loader/leap_binder/leapbinder.py,sha256=WjC-UA5xeY9UZtMqJQOxjidlFcWURC9mmy7OrQy1IY0,6692
-code_loader/leap_loader_parallelized_base.py,sha256=3lnYq-6FwnhVFD4gJ3maOdK_c8huM8pEdfze1CZnhAU,4507
+code_loader/leap_loader_parallelized_base.py,sha256=K9fuC4k06z7xpLO2p_THW0D3uw1-N6yTK7h3DP3YSFU,4436
 code_loader/leaploader.py,sha256=MnQeNOwNgkQItCWFiWT6xBxAsqjNbcTOPk8HIP8v19o,15370
-code_loader/metric_calculator_parallelized.py,sha256=7AWBHQVs_lhRLsLFN861sWlaviGqx0l-iGOn0jXj53Y,2075
+code_loader/metric_calculator_parallelized.py,sha256=lERwpXnv7IF-mJ3_5hujIKRPUpWtzaL5-RQkoCxEBo8,2128
 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=Sv4PZmpFsiqMm9kuJR9Llcl094W88h_kXUTHBbcnby8,2335
 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.84.dev1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-0.2.84.dev1.dist-info/METADATA,sha256=70BTlqi8zQDCK-UEcZ8hxTuq5Qik-jvhMSkaDZrFrIY,952
-code_loader-0.2.84.dev1.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-code_loader-0.2.84.dev1.dist-info/RECORD,,
+code_loader-0.2.84.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-0.2.84.dev3.dist-info/METADATA,sha256=2GYrf1LNZ108TvfdYpb2Ec1I43zm_zjLUkQk2NgGPPU,952
+code_loader-0.2.84.dev3.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+code_loader-0.2.84.dev3.dist-info/RECORD,,