empty arg constraints

cai4cai · Aug 15, 2023 · 73e827d · 73e827d
1 parent a36b9b1
commit 73e827d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchsparsegradutils/distributions/sparse_multivariate_normal.py b/torchsparsegradutils/distributions/sparse_multivariate_normal.py
@@ -76,6 +76,7 @@ class SparseMultivariateNormal(Distribution):
             in either torch.sparse_coo or torch.sparse_csr layout
     """
 
+    arg_contraints = {}
     # TODO: add in constraints
     # arg_constraints = {'loc': constraints.real_vector,
     #                    'diag': constraints.independent(constraints.positive, 1),