code-loader 1.0.99.dev9__tar.gz → 1.0.99.dev10__tar.gz

{code_loader-1.0.99.dev9 → code_loader-1.0.99.dev10}/PKG-INFO

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

{code_loader-1.0.99.dev9 → code_loader-1.0.99.dev10}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -44,6 +44,23 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
         class TempMapping:
             pass
 
+        def inner():
+            class ModelPlaceholder:
+                def __init__(self):
+                    self.model = load_model_func()
+                #keras interface
+                def __call__(self, arg):
+                    ret = self.model(arg)
+                    return ret.numpy()
+
+                # onnx runtime interface
+                def run(self, output_names, input_dict):
+                    return self.model.run(output_names, input_dict)
+
+
+            return ModelPlaceholder()
+
+
         def mapping_inner():
             class ModelOutputPlaceholder:
                 def __init__(self):
@@ -81,11 +98,13 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
             return ModelPlaceholder()
 
 
+
+
         def final_inner():
             if os.environ.get(mapping_runtime_mode_env_var_mame):
                 return mapping_inner()
             else:
-                return load_model_func()
+                return inner()
 
         return final_inner
 

{code_loader-1.0.99.dev9 → code_loader-1.0.99.dev10}/pyproject.toml

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