code-loader 1.0.140__tar.gz → 1.0.141__tar.gz

{code_loader-1.0.140 → code_loader-1.0.141}/PKG-INFO

@@ -1,9 +1,9 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.140
+Version: 1.0.141
 Summary: 
+Home-page: https://github.com/tensorleap/code-loader
 License: MIT
-License-File: LICENSE
 Author: dorhar
 Author-email: doron.harnoy@tensorleap.ai
 Requires-Python: >=3.8,<3.13
@@ -20,7 +20,6 @@ Requires-Dist: numpy (>=2.3.2,<3.0.0) ; python_version >= "3.11" and python_vers
 Requires-Dist: psutil (>=5.9.5,<6.0.0)
 Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
-Project-URL: Homepage, https://github.com/tensorleap/code-loader
 Project-URL: Repository, https://github.com/tensorleap/code-loader
 Description-Content-Type: text/markdown
 

{code_loader-1.0.140 → code_loader-1.0.141}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -122,6 +122,9 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                 # keras interface
                 def __call__(self, arg):
                     ret = self.model(arg)
+                    if isinstance(ret, list or tuple):
+                        return [r.numpy() for r in ret]
+
                     return ret.numpy()
 
                 def _convert_onnx_inputs_to_correct_type(
@@ -657,7 +660,7 @@ def tensorleap_preprocess():
 
             result = user_function()
             _validate_result(result)
-            
+
             # Emit integration test event once per test
             try:
                 emit_integration_event_once(AnalyticsEvent.PREPROCESS_INTEGRATION_TEST, {
@@ -665,7 +668,7 @@ def tensorleap_preprocess():
                 })
             except Exception as e:
                 logger.debug(f"Failed to emit preprocess integration test event: {e}")
-            
+
             return result
 
         return inner

{code_loader-1.0.140 → code_loader-1.0.141}/code_loader/plot_functions/plot_functions.py

@@ -324,7 +324,7 @@ def plot_image_mask(leap_data: LeapImageMask, title: str) -> None:
 
         # fill the instance mask with a translucent color
         overlayed_image[instance_mask] = (
-                overlayed_image[instance_mask] * (1 - 0.5) + np.array(colors[i][:image.shape[-1]], dtype=np.uint8) * 0.5)
+                overlayed_image[instance_mask] * (1 - 0.5) + np.array(colors[i][:image.shape[-1]], dtype=image.dtype) * 0.5)
 
     # Display the result using matplotlib
     fig, ax = plt.subplots(1)

{code_loader-1.0.140 → code_loader-1.0.141}/pyproject.toml

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