sagemaker-python-sdk
sagemaker-python-sdk copied to clipboard
More typing in the codebase, specifically for FrameworkProcessors
Describe the feature you'd like I want type validations when I am constructing FrameworkProcessors and calling their run method.
How would this feature be used? Please describe. It will help developers understand what the function is doing by providing the type information. There is some type information in the docstring of the FrameworkProcessor, but this docstring is not available from a subclass like PytorchProcessor. It should be easier to get the type signature of these functions.
Additional context I don't think the scope of the issue should be the entire codebase. I am limiting this issue to the FrameworkProcessors so that I can submit a pull request to resolve it myself.