code-loader 0.2.90__tar.gz → 0.2.90.dev31__tar.gz

{code_loader-0.2.90 → code_loader-0.2.90.dev31}/PKG-INFO

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

{code_loader-0.2.90 → code_loader-0.2.90.dev31}/code_loader/metric_calculator_parallelized.py

@@ -4,7 +4,7 @@ from multiprocessing import Process, Queue
 from code_loader.leap_loader_parallelized_base import LeapLoaderParallelizedBase
 import traceback
 from dataclasses import dataclass
-import tensorflow as tf
+import numpy as np
 from code_loader.leaploader import LeapLoader
 
 
@@ -31,8 +31,7 @@ class MetricCalculatorParallelized(LeapLoaderParallelizedBase):
         while True:
             metric_id, metric_name, input_arg_name_to_tensor = metrics_to_process.get(block=True)
             try:
-                with tf.device('/cpu:0'):
-                    metric_result = leap_loader.metric_by_name()[metric_name].function(**input_arg_name_to_tensor)
+                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)))
@@ -49,5 +48,5 @@ class MetricCalculatorParallelized(LeapLoaderParallelizedBase):
         process.start()
         return process
 
-    def calculate_metrics(self, input_arg_name_to_tensor_list: List[Tuple[str, str, Dict[str, tf.Tensor]]]):
+    def calculate_metrics(self, input_arg_name_to_tensor_list: List[Tuple[str, str, Dict[str, np.array]]]):
         return self.start_process_inputs(input_arg_name_to_tensor_list)

{code_loader-0.2.90 → code_loader-0.2.90.dev31}/code_loader/visualizer_calculator_parallelized.py

@@ -6,7 +6,6 @@ import numpy as np
 
 from code_loader.leap_loader_parallelized_base import LeapLoaderParallelizedBase
 from dataclasses import dataclass
-import tensorflow as tf
 from code_loader.leaploader import LeapLoader
 
 
@@ -40,9 +39,8 @@ class VisualizerCalculatorParallelized(LeapLoaderParallelizedBase):
             index_in_batch, visualizer_id, visualizer_name, input_arg_name_to_tensor = \
                 visualizers_to_process.get(block=True)
             try:
-                with tf.device('/cpu:0'):
-                    visualizer_result = \
-                        leap_loader.visualizer_by_name()[visualizer_name].function(**input_arg_name_to_tensor)
+                visualizer_result = \
+                    leap_loader.visualizer_by_name()[visualizer_name].function(**input_arg_name_to_tensor)
             except Exception as e:
                 ready_visualizations.put(VisualizerSerializableError(
                     visualizer_id, visualizer_name, index_in_batch, str(e)))

{code_loader-0.2.90 → code_loader-0.2.90.dev31}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "0.2.90"
+version = "0.2.90.dev31"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"