Skip to content

Commit 47d9f7f

Browse files
committed
bugfix in nplets_measures when no nplets passed as parameter
1 parent c9a6b53 commit 47d9f7f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

thoi/measures/gaussian_copula.py

+2
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,8 @@ def nplets_measures(X: Union[TensorLikeArray],
192192
logging.warning('Using hot encoding to compute multi-order measures as nplets have different orders')
193193
nplets = _indices_to_hot_encoded(nplets, N)
194194
return nplets_measures_hot_encoded(covmats, nplets, covmat_precomputed=True, T=T)
195+
elif nplets is None:
196+
nplets = torch.arange(N, device=device).unsqueeze(0)
195197

196198
# If nplets are not tensors, convert to tensor
197199
nplets = torch.as_tensor(nplets).to(device).contiguous()

0 commit comments

Comments
 (0)