Skip to content

Commit 9069567

Browse files
fix
1 parent 366bdb8 commit 9069567

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

GNNGraphs/src/convert.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ function to_dense(A::ADJMAT_T, T = nothing; dir = :out, num_nodes = nothing,
137137
A = T.(A)
138138
end
139139
if !weighted
140-
A = map(x -> ifelse(x > 0, T(1), T(0)), A)
140+
A = binarize(A, T)
141141
end
142142
return A, num_nodes, num_edges
143143
end

GNNlib/test/runtests.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ using TestItemRunner
1313

1414
## Uncomment below and in test_module.jl to change the default test settings
1515
# ENV["GNN_TEST_CPU"] = "false"
16-
# ENV["GNN_TEST_CUDA"] = "true"
16+
ENV["GNN_TEST_CUDA"] = "true"
1717
# ENV["GNN_TEST_AMDGPU"] = "true"
1818
# ENV["GNN_TEST_Metal"] = "true"
1919

GNNlib/test/test_module.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ using Pkg
66
# tried to put this in __init__ but is not executed for some reason
77

88
## Uncomment below to change the default test settings
9-
# ENV["GNN_TEST_CUDA"] = "true"
9+
ENV["GNN_TEST_CUDA"] = "true"
1010
# ENV["GNN_TEST_AMDGPU"] = "true"
1111
# ENV["GNN_TEST_Metal"] = "true"
1212

0 commit comments

Comments
 (0)