From 39aed1e1f5148d852c73f1ef2a202ef8fab74b89 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Mon, 10 Jun 2024 23:03:49 -0700 Subject: [PATCH] Fix docs --- Project.toml | 2 +- src/layers/hamiltonian.jl | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Project.toml b/Project.toml index 82241d9..37f5ea6 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "Boltz" uuid = "4544d5e4-abc5-4dea-817f-29e4c205d9c8" authors = ["Avik Pal and contributors"] -version = "0.3.6" +version = "0.3.7" [deps] ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b" diff --git a/src/layers/hamiltonian.jl b/src/layers/hamiltonian.jl index 2656c71..944d9ba 100644 --- a/src/layers/hamiltonian.jl +++ b/src/layers/hamiltonian.jl @@ -10,7 +10,7 @@ position and momentum. ## Arguments - `FST`: If `true`, then the type of the state returned by the model must be same as the - type of the input state. See the documentation on [`StatefulLayer`](@ref) for more + type of the input state. See the documentation on [`StatefulLuxLayer`](@ref) for more information. - `model`: A `Lux.AbstractExplicitLayer` neural network that returns the Hamiltonian of the system. The `model` must return a "batched scalar", i.e. all the dimensions of the