Skip to content

Commit c449083

Browse files
committed
add some comments for distr
1 parent c869da0 commit c449083

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/uni2ts/distribution/_base.py

+5
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,11 @@ def variance(self) -> torch.Tensor:
132132

133133
@abstract_class_property("distr_cls")
134134
class DistributionOutput:
135+
"""
136+
Base class for distribution outputs.
137+
Defines the type of output distribution and provides several helper methods for predictive distributions.
138+
"""
139+
135140
distr_cls: type[Distribution] = NotImplemented
136141

137142
def distribution(

0 commit comments

Comments
 (0)