code-loader 1.0.61.dev5__py3-none-any.whl → 1.0.61.dev7__py3-none-any.whl

code_loader/leaploader.py

@@ -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/leaploaderbase.py

@@ -57,6 +57,11 @@ class LeapLoaderBase:
                    input_tensors_by_arg_name: Dict[str, npt.NDArray[np.float32]]) -> MetricCallableReturnType:
         pass
 
+    @abstractmethod
+    def run_custom_loss(self, custom_loss_name: str,
+                        input_tensors_by_arg_name: Dict[str, npt.NDArray[np.float32]]):
+        pass
+
     @abstractmethod
     def run_heatmap_visualizer(self, visualizer_name: str, input_tensors_by_arg_name: Dict[str, npt.NDArray[np.float32]]
                                ) -> npt.NDArray[np.float32]:

{code_loader-1.0.61.dev5.dist-info → code_loader-1.0.61.dev7.dist-info}/METADATA

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

{code_loader-1.0.61.dev5.dist-info → code_loader-1.0.61.dev7.dist-info}/RECORD

@@ -20,12 +20,12 @@ code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaS
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=u7WF_nfXTe2c5pvcgVbBzJEf61zGzTGlhPNBa0usaVE,25226
 code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=uuM_ht9HZ1GH2IabKeGQ_x9NmD3poK_h1Gt0NruwJuY,19704
-code_loader/leaploader.py,sha256=obFJFRWM4PzKlWcW7JrB41NmFJkyJuqiJOD8tcp4e7Y,21332
-code_loader/leaploaderbase.py,sha256=Qh86EzeJ8YzLhSjzjhMDUARKwTp8z3-X1BlP5MHS3BY,2761
+code_loader/leaploader.py,sha256=Cwc8jnG_AUEC1QN9AyzEsftkCcMaTcY-JqrDKJzkWS4,21571
+code_loader/leaploaderbase.py,sha256=waxwf5xQP8FhJjP7v7kRxDIRzyGu_26Z-glHWXyDt4g,2936
 code_loader/utils.py,sha256=aw2i_fqW_ADjLB66FWZd9DfpCQ7mPdMyauROC5Nd51I,2197
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=VoqO9FN84yXyMjRjHjUTOt2GdTkJRMbHbXJ1cJkREkk,2230
-code_loader-1.0.61.dev5.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.61.dev5.dist-info/METADATA,sha256=pCYB-NtpNjx41VFMyJUMEoOh6JQrJFHT2Dnl7mmwmE4,893
-code_loader-1.0.61.dev5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.61.dev5.dist-info/RECORD,,
+code_loader-1.0.61.dev7.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.61.dev7.dist-info/METADATA,sha256=PnP3GQ_WtQt9i9McMeLePnK2TP-WongXzRT1RID9Zg4,893
+code_loader-1.0.61.dev7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.61.dev7.dist-info/RECORD,,