Expose GetSessionOptions in pybind logic and add unit test for python

Author: hariharans29

onnxruntime/python/onnxruntime_pybind_state.cc

@@ -779,6 +779,9 @@ including arg name, arg type (contains both type and shape).)pbdoc")
       .def("get_providers", [](InferenceSession* sess) -> const std::vector<std::string>& {
         return sess->GetRegisteredProviderTypes();
       })
+      .def_property_readonly("session_options", [](InferenceSession* sess) -> const SessionOptions& {
+        return sess->GetSessionOptions();
+      })
       .def_property_readonly("inputs_meta", [](const InferenceSession* sess) -> const std::vector<const onnxruntime::NodeArg*>& {
         auto res = sess->GetModelInputs();
         OrtPybindThrowIfError(res.first);

onnxruntime/python/session.py

@@ -40,6 +40,7 @@ def _load_model(self, providers=[]):
 
         self._sess.load_model(providers)
 
+        self._session_options = self._sess.session_options
         self._inputs_meta = self._sess.inputs_meta
         self._outputs_meta = self._sess.outputs_meta
         self._overridable_initializers = self._sess.overridable_initializers
@@ -63,6 +64,10 @@ def _reset_session(self):
         self._providers = None
         self._sess = None
 
+    def get_session_options(self):
+        "Return the session options. See :class:`onnxruntime.SessionOptions`."
+        return self._session_options
+
     def get_inputs(self):
         "Return the inputs metadata as a list of :class:`onnxruntime.NodeArg`."
         return self._inputs_meta

onnxruntime/test/python/onnxruntime_test_python.py

@@ -594,5 +594,28 @@ def testOrtExecutionMode(self):
         opt.execution_mode = onnxrt.ExecutionMode.ORT_PARALLEL
         self.assertEqual(opt.execution_mode, onnxrt.ExecutionMode.ORT_PARALLEL)
 
+    def testLoadingSessionOptionsFromModel(self):
+        try:
+            os.environ['ORT_LOAD_CONFIG_FROM_MODEL'] = str(1)
+            sess = onnxrt.InferenceSession(self.get_name("model_with_valid_ort_config_json.onnx"))
+            session_options = sess.get_session_options()
+            
+            self.assertEqual(session_options.inter_op_num_threads, 5)
+            
+            self.assertEqual(session_options.intra_op_num_threads, 2)
+
+            self.assertEqual(session_options.execution_mode, onnxrt.ExecutionMode.ORT_PARALLEL)
+
+            self.assertEqual(session_options.graph_optimization_level, 3)
+
+            self.assertEqual(session_options.enable_profiling, True)
+            
+        except Exception: 
+            raise
+
+        finally:
+            # Make sure the usage of the feature is disabled after this test
+            os.environ['ORT_LOAD_CONFIG_FROM_MODEL'] = str(0)       
+        
 if __name__ == '__main__':
     unittest.main()