diff --git a/torchsparsegradutils/distributions/sparse_multivariate_normal.py b/torchsparsegradutils/distributions/sparse_multivariate_normal.py index f69e201..492c0a5 100644 --- a/torchsparsegradutils/distributions/sparse_multivariate_normal.py +++ b/torchsparsegradutils/distributions/sparse_multivariate_normal.py @@ -76,7 +76,7 @@ class SparseMultivariateNormal(Distribution): in either torch.sparse_coo or torch.sparse_csr layout """ - arg_contraints = {} + arg_constraints = {} # TODO: add in constraints # arg_constraints = {'loc': constraints.real_vector, # 'diag': constraints.independent(constraints.positive, 1),