single_dispatch.py

from functools import singledispatch
import numpy as np

@singledispatch
def test_function(arg):
  print("default")

@test_function.register(int)
def _(arg):
  print(f"arg is integer {arg}")
  
@test_function.register(str)
def _(arg):
  print(f"arg is string {arg}")
  
@test_function.register(np.ndarray)
def _(arg):
  print(f"arg is ndarray {arg}")