[no ci] docs: manually add types to types module docstring

Types are not recognized by autodoc, so it seem we have to add them by hand

Author: vpratz

bayesflow/types/__init__.py

@@ -1,3 +1,12 @@
+"""
+Custom types.
+
+.. currentmodule:: bayesflow.types
+
+.. autodata:: Shape
+.. autodata:: Tensor
+"""
+
 from .shape import Shape
 from .tensor import Tensor