Type-checking issues have surfaced

[nukularrr]

Although the python source code hasn't changed, pyright is now surfacing type issues, e.g.:

/home/runner/work/responsible-ai-toolbox/responsible-ai-toolbox/tests/annotations.py:34:42 - error: Argument of type "Type[SGD]" cannot be assigned to parameter "InnerOpt" of type "Partial[Optimizer] | OptimizerType" in function "__init__" Type "Type[SGD]" cannot be assigned to type "Partial[Optimizer] | OptimizerType" Type "Type[SGD]" cannot be assigned to type "(*args: Any, **kwargs: Any) -> Optimizer" Parameter "*args" has no corresponding parameter Parameter "**kwargs" has no corresponding parameter Function return type "SGD" is incompatible with type "Optimizer" "SGD" is incompatible with protocol "Optimizer" "step" is an incompatible type Type "Type[SGD]" cannot be assigned to type "OptimizerType" ... (reportGeneralTypeIssues)