code-loader 1.0.61.dev6__tar.gz → 1.0.61.dev8__tar.gz

{code_loader-1.0.61.dev6 → code_loader-1.0.61.dev8}/PKG-INFO

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

{code_loader-1.0.61.dev6 → code_loader-1.0.61.dev8}/code_loader/leaploader.py

@@ -70,7 +70,7 @@ class LeapLoader(LeapLoaderBase):
         self.exec_script()
         setup = global_leap_binder.setup_container
         return {
-            metric_handler.name: metric_handler.metric_handler_data
+            metric_handler.metric_handler_data.name: metric_handler.metric_handler_data
             for metric_handler in setup.metrics
         }
 
@@ -79,7 +79,7 @@ class LeapLoader(LeapLoaderBase):
         self.exec_script()
         setup = global_leap_binder.setup_container
         return {
-            metric_handler.name: metric_handler
+            metric_handler.metric_handler_data.name: metric_handler
             for metric_handler in setup.metrics
         }
 
@@ -88,7 +88,7 @@ class LeapLoader(LeapLoaderBase):
         self.exec_script()
         setup = global_leap_binder.setup_container
         return {
-            visualizer_handler.name: visualizer_handler.visualizer_handler_data
+            visualizer_handler.visualizer_handler_data.name: visualizer_handler.visualizer_handler_data
             for visualizer_handler in setup.visualizers
         }
 
@@ -97,7 +97,7 @@ class LeapLoader(LeapLoaderBase):
         self.exec_script()
         setup = global_leap_binder.setup_container
         return {
-            visualizer_handler.name: visualizer_handler
+            visualizer_handler.visualizer_handler_data.name: visualizer_handler
             for visualizer_handler in setup.visualizers
         }
 
@@ -106,7 +106,7 @@ class LeapLoader(LeapLoaderBase):
         self.exec_script()
         setup = global_leap_binder.setup_container
         return {
-            custom_loss_handler.name: custom_loss_handler.custom_loss_handler_data
+            custom_loss_handler.custom_loss_handler_data.name: custom_loss_handler.custom_loss_handler_data
             for custom_loss_handler in setup.custom_loss_handlers
         }
 
@@ -115,7 +115,7 @@ class LeapLoader(LeapLoaderBase):
         self.exec_script()
         setup = global_leap_binder.setup_container
         return {
-            custom_loss_handler.name: custom_loss_handler
+            custom_loss_handler.custom_loss_handler_data.name: custom_loss_handler
             for custom_loss_handler in setup.custom_loss_handlers
         }
 
@@ -300,10 +300,13 @@ class LeapLoader(LeapLoaderBase):
                                                               type=dataset_metadata_type))
 
         visualizers = [
-            VisualizerInstance(visualizer_handler.name, visualizer_handler.type, visualizer_handler.arg_names)
+            VisualizerInstance(
+                visualizer_handler.visualizer_handler_data.name, visualizer_handler.visualizer_handler_data.type,
+                visualizer_handler.visualizer_handler_data.arg_names)
             for visualizer_handler in setup.visualizers]
 
-        custom_losses = [CustomLossInstance(custom_loss.name, custom_loss.arg_names)
+        custom_losses = [CustomLossInstance(custom_loss.custom_loss_handler_data.name,
+                                            custom_loss.custom_loss_handler_data.arg_names)
                          for custom_loss in setup.custom_loss_handlers]
 
         prediction_types = []
@@ -314,7 +317,7 @@ class LeapLoader(LeapLoaderBase):
 
         metrics = []
         for metric in setup.metrics:
-            metric_inst = MetricInstance(metric.name, metric.arg_names)
+            metric_inst = MetricInstance(metric.metric_handler_data.name, metric.metric_handler_data.arg_names)
             metrics.append(metric_inst)
 
         return DatasetSetup(preprocess=dataset_preprocess, inputs=inputs, outputs=ground_truths,

{code_loader-1.0.61.dev6 → code_loader-1.0.61.dev8}/pyproject.toml

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