Add smarttypeX to apply intX or floatX as needed

This will be used to fix the `GeneratorAdapter` when applied
to generators producing int-valued data.

Author: michaelosthege

docs/source/api/pytensorf.rst

@@ -16,6 +16,7 @@ PyTensor utils
    floatX
    intX
    smartfloatX
+   smarttypeX
    constant_fold
    CallableTensor
    join_nonshared_inputs

pymc/pytensorf.py

@@ -68,6 +68,7 @@
     "floatX",
     "intX",
     "smartfloatX",
+    "smarttypeX",
     "jacobian",
     "CallableTensor",
     "join_nonshared_inputs",
@@ -297,6 +298,14 @@ def smartfloatX(x):
     return x
 
 
+def smarttypeX(x):
+    if str(x.dtype).startswith("float"):
+        x = floatX(x)
+    elif str(x.dtype).startswith("int"):
+        x = intX(x)
+    return x
+
+
 """
 PyTensor derivative functions
 """