diff --git a/src/autora/variable/__init__.py b/src/autora/variable/__init__.py index f42337ac..68906efd 100644 --- a/src/autora/variable/__init__.py +++ b/src/autora/variable/__init__.py @@ -16,6 +16,7 @@ class ValueType(str, Enum): "probability_distribution" # probability distribution over classes ) CLASS = "class" # sample from probability distribution over classes + MATRIX = "matrix" # matrix of probabilities @dataclass