Add get_model method to ONNXRuntimeAdapter

ashwinvaidya17 · ashwinvaidya17 · commit 7d33688fe5fc · 2024-10-31T13:10:56.000+01:00
Signed-off-by: Ashwin Vaidya &lt;ashwinnitinvaidya@gmail.com&gt;
diff --git a/model_api/python/model_api/adapters/onnx_adapter.py b/model_api/python/model_api/adapters/onnx_adapter.py
@@ -162,6 +162,10 @@ def embed_preprocessing(
             reversed(preproc_funcs),
         )
 
+    def get_model(self):
+        """Return the reference to the ONNXRuntime session."""
+        return self.session
+
     def reshape_model(self, new_shape):
         raise NotImplementedError
 

Original file line number	Diff line number	Diff line change
`@@ -162,6 +162,10 @@ def embed_preprocessing(`
`162`	`162`	`reversed(preproc_funcs),`
`163`	`163`	`)`
`164`	`164`
	`165`	`+ def get_model(self):`
	`166`	`+ """Return the reference to the ONNXRuntime session."""`
	`167`	`+ return self.session`
	`168`	`+`
`165`	`169`	`def reshape_model(self, new_shape):`
`166`	`170`	`raise NotImplementedError`
`167`	`171`