onnxruntime/orttraining/orttraining/python
pengwa 34fe8fb069
Type hint for ORTModule (#15938)
### Type hint for ORTModule

Add Type hint for ORTModule
Refine comments. 

The reason of removing theinterface execution_session_run_forward from
`orttraining/orttraining/python/training/ortmodule/_graph_execution_manager.py`:

PR
cc275e7529 (diff-497e18dc8878818205b81fd80f85942548d8aa15d0f1204ce3e3d9795e3dd195)
and some commit before it breaks the function interface contracts
between parent calss _graph_execution_manager.py and its children
_training_manager.py and _inference_manager.py. So there is no need to
have this interface.


### Other EE work opportunities

1. Use logger correctly. 
2. Remove few duplication logic parsing input/output recursively.
3. Clean up environment variable usage.
2023-05-25 09:28:20 +08:00
..
deprecated Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
training Type hint for ORTModule (#15938) 2023-05-25 09:28:20 +08:00
checkpointing_utils.py Bump ruff in CI (#15533) 2023-04-17 10:11:44 -07:00
ort_trainer.py Bump ruff in CI (#15533) 2023-04-17 10:11:44 -07:00
orttraining_pybind_common.h Re-work global objects dependancies in pybind layer. (#14941) 2023-03-10 13:55:31 -08:00
orttraining_pybind_state.cc Enable conditional optimization automatically (#15885) 2023-05-23 13:08:05 +08:00
orttraining_python_module.cc Expose build information in dynamic lib (#15643) 2023-04-28 21:57:31 -07:00
orttraining_python_module_eager.h Run clang-format in CI (#15524) 2023-04-18 09:26:58 -07:00
pt_patch.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00