Skip to content

Commit 57670c8

Browse files
committed
make check_invariants explicit
1 parent 12183c8 commit 57670c8

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

python/coverage_control/coverage_env_utils.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -203,11 +203,15 @@ def get_communication_maps(
203203
values = neighbors_pos / params.pCommunicationRange
204204
# values = values / params.pCommunicationRange
205205
# values = (values + params.pCommunicationRange) / (2. * params.pCommunicationRange)
206+
# Invariant checks are off by default; opting out explicitly
207+
# silences the UserWarning torch emits on first use.
206208
comm_maps[r_idx][0] = torch.sparse_coo_tensor(
207-
indices, values[:, 0], torch.Size([map_size, map_size])
209+
indices, values[:, 0], torch.Size([map_size, map_size]),
210+
check_invariants=False,
208211
).to_dense()
209212
comm_maps[r_idx][1] = torch.sparse_coo_tensor(
210-
indices, values[:, 1], torch.Size([map_size, map_size])
213+
indices, values[:, 1], torch.Size([map_size, map_size]),
214+
check_invariants=False,
211215
).to_dense()
212216

213217
return comm_maps

0 commit comments

Comments
 (0)