import Flux

JuliaGraphs · Jul 22, 2024 · dc99dd3 · dc99dd3
1 parent 0e8d8f5
commit dc99dd3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/GNNGraphs/GNNGraphs.jl b/src/GNNGraphs/GNNGraphs.jl
@@ -7,6 +7,8 @@ using Graphs: AbstractGraph, outneighbors, inneighbors, adjacency_matrix, degree
               has_self_loops, is_directed
 import NearestNeighbors
 import NNlib
+import Flux
+using Flux: batch
 import StatsBase
 import KrylovKit
 using ChainRulesCore