Skip to content

Commit

Permalink
Test documentation for variable
Browse files Browse the repository at this point in the history
  • Loading branch information
bdvllrs committed Mar 5, 2024
1 parent 12edc51 commit 59628e3
Showing 1 changed file with 12 additions and 9 deletions.
21 changes: 12 additions & 9 deletions shimmer/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,11 @@

import torch

test = ""
"""Does this show ?"""

RawDomainGroupT = Mapping[str, Any]
RawDomainGroupT.__doc__ = """
"""
Matched raw unimodal data from multiple domains.
Keys of the mapping are domains names.
Expand All @@ -14,7 +17,7 @@
"""

RawDomainGroupDT = dict[str, Any]
RawDomainGroupDT.__doc__ = """
"""
Matched raw unimodal data from multiple domains.
Keys of the dict are domains names.
Expand All @@ -26,7 +29,7 @@
"""

LatentsDomainGroupT = Mapping[str, torch.Tensor]
LatentsDomainGroupT.__doc__ = """
"""
Matched unimodal latent representations from multiple domains.
Keys of the mapping are domains names.
Expand All @@ -36,7 +39,7 @@
"""

LatentsDomainGroupDT = dict[str, torch.Tensor]
LatentsDomainGroupDT.__doc__ = """
"""
Matched unimodal latent representations from multiple domains.
Keys of the dict are domains names.
Expand All @@ -48,7 +51,7 @@
"""

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).
Expand All @@ -58,7 +61,7 @@
"""

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).
Expand All @@ -71,7 +74,7 @@
"""

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).
Expand All @@ -81,7 +84,7 @@
"""

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).
Expand All @@ -93,7 +96,7 @@
"""

ModelModeT = Literal["train", "val", "test", "val/ood", "test/ood"]
ModelModeT.__doc__ = """Mode used by pytorch lightning.
"""Mode used by pytorch lightning.
```python
ModelModeT = Literal["train", "val", "test", "val/ood", "test/ood"]
Expand Down

0 comments on commit 59628e3

Please sign in to comment.