MCPcopy
hub / github.com/pyg-team/pytorch_geometric / jit

Function jit

test/utils/test_undirected.py:23–24  ·  view source on GitHub ↗
(edge_index: Tensor)

Source from the content-addressed store, hash-verified

21
22 @torch.jit.script
23 def jit(edge_index: Tensor) -> bool:
24 return is_undirected(edge_index)
25
26 assert not jit(torch.stack([row, col], dim=0))
27

Callers 15

test_contains_isolated_nodesFunction · 0.85
test_remove_isolated_nodesFunction · 0.85
test_softmaxFunction · 0.85
test_scatterFunction · 0.85
test_get_laplacianFunction · 0.85
test_is_undirectedFunction · 0.85
test_to_undirectedFunction · 0.85
test_gridFunction · 0.85
test_normalized_cutFunction · 0.85
test_to_dense_adjFunction · 0.85
test_dense_to_sparseFunction · 0.85
test_to_torch_coo_tensorFunction · 0.85

Calls 2

is_undirectedFunction · 0.90
to_undirectedFunction · 0.90

Tested by

no test coverage detected