Add docstring for type definitions. Pdoc does not detect them automat…

…ically.

shimmer/types.py

@@ -4,61 +4,98 @@
 import torch
 
 RawDomainGroupT = Mapping[str, Any]
-"""
+RawDomainGroupT.__doc__ = """
 Matched raw unimodal data from multiple domains.
 Keys of the mapping are domains names.
+
+```python
+    RawDomainGroupT = Mapping[str, Any]
+```
 """
 
 RawDomainGroupDT = dict[str, Any]
-"""
+RawDomainGroupDT.__doc__ = """
 Matched raw unimodal data from multiple domains.
 Keys of the dict are domains names.
 
 This is a more specific version of `RawDomainGroupT` used in method's outputs.
+
+```python
+    RawDomainGroupDT = dict[str, Any]
+```
 """
 
 LatentsDomainGroupT = Mapping[str, torch.Tensor]
-"""
+LatentsDomainGroupT.__doc__ = """
 Matched unimodal latent representations from multiple domains.
 Keys of the mapping are domains names.
+
+```python
+    LatentsDomainGroupT = Mapping[str, torch.Tensor]
+```
 """
 
 LatentsDomainGroupDT = dict[str, torch.Tensor]
-"""
+LatentsDomainGroupDT.__doc__ = """
 Matched unimodal latent representations from multiple domains.
 Keys of the dict are domains names.
 
 This is a more specific version of `LatentsDomainGroupT` used in method's outputs.
+
+```python
+    LatentsDomainGroupDT = dict[str, torch.Tensor]
+```
 """
 
 LatentsDomainGroupsT = Mapping[frozenset[str], LatentsDomainGroupT]
-"""
+LatentsDomainGroupsT.__doc__ = """
 Mapping of `LatentsDomainGroupT`. Keys are frozenset of domains matched in the group.
 Each group is independent and contains different data (unpaired).
+
+```python
+    LatentsDomainGroupsT = Mapping[frozenset[str], LatentsDomainGroupT]
+```
 """
 
 LatentsDomainGroupsDT = dict[frozenset[str], LatentsDomainGroupDT]
-"""
+LatentsDomainGroupsDT.__doc__ = """
 Mapping of `LatentsDomainGroupDT`.
 Keys are frozenset of domains matched in the group.
 Each group is independent and contains different data (unpaired).
 
 This is a more specific version of `LatentsDomainGroupsT` used in method's outputs.
+
+```python
+    LatentsDomainGroupsDT = dict[frozenset[str], LatentsDomainGroupDT]
+```
 """
 
 RawDomainGroupsT = Mapping[frozenset[str], RawDomainGroupT]
-"""
+RawDomainGroupsT.__doc__ = """
 Mapping of `RawDomainGroupT`. Keys are frozenset of domains matched in the group.
 Each group is independent and contains different data (unpaired).
+
+```python
+    RawDomainGroupsT = Mapping[frozenset[str], RawDomainGroupT]
+```
 """
 
 RawDomainGroupsDT = dict[frozenset[str], RawDomainGroupDT]
-"""
+RawDomainGroupsDT.__doc__ = """
 Mapping of `RawDomainGroupT`. Keys are frozenset of domains matched in the group.
 Each group is independent and contains different data (unpaired).
 
 This is a more specific version of `RawDomainGroupsT` used in method's outputs.
+
+```python
+    RawDomainGroupsDT = dict[frozenset[str], RawDomainGroupDT]
+```
 """
 
 ModelModeT = Literal["train", "val", "test", "val/ood", "test/ood"]
-""" Mode used by pytorch lightning."""
+ModelModeT.__doc__ = """Mode used by pytorch lightning.
+
+```python
+    ModelModeT = Literal["train", "val", "test", "val/ood", "test/ood"]
+```
+"""