I wonder what’s the right way to type annotate a python function as below? This is a toy example that is analogous to a python I encountered that needs type annotated. Using Union like doesn’t feel right because it doesn’t enforce the rules an int input must result in an int output. a float input must result in a float