fix docs

FluxML · Oct 10, 2024 · 2d91378 · 2d91378
1 parent 3680c23
commit 2d91378
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/make.jl b/docs/make.jl
@@ -1,11 +1,11 @@
 using Documenter, Flux, NNlib, Functors, MLUtils, BSON, Optimisers, 
       OneHotArrays, Zygote, ChainRulesCore, Plots, MLDatasets, Statistics, 
-      DataFrames, JLD2
+      DataFrames, JLD2, MLDataDevices
 
 DocMeta.setdocmeta!(Flux, :DocTestSetup, :(using Flux); recursive = true)
 
 makedocs(
-    modules = [Flux, NNlib, Functors, MLUtils, Zygote, OneHotArrays, Optimisers, ChainRulesCore],
+    modules = [Flux, NNlib, Functors, MLUtils, Zygote, OneHotArrays, Optimisers, ChainRulesCore, MLDataDevices],
     sitename = "Flux",
     pages = [
         "Welcome" => "index.md",