From 933fc27f8666f4c72b2d9a70c7d50e9d93b430c0 Mon Sep 17 00:00:00 2001
From: Michael Goerz <mail@michaelgoerz.net>
Date: Sun, 21 Apr 2024 23:30:52 -0400
Subject: [PATCH] Add support for `get_parameters`

---
 src/grad_generator.jl | 6 ++++++
 test/Project.toml     | 1 +
 2 files changed, 7 insertions(+)

diff --git a/src/grad_generator.jl b/src/grad_generator.jl
index 3ed30ba..0a5cdc7 100644
--- a/src/grad_generator.jl
+++ b/src/grad_generator.jl
@@ -1,5 +1,6 @@
 using QuantumControlBase: get_control_derivs
 import QuantumControlBase.QuantumPropagators.Controls: get_controls
+import QuantumControlBase.QuantumPropagators.Controls: get_parameters
 import QuantumControlBase.QuantumPropagators: _exp_prop_convert_operator
 
 
@@ -50,4 +51,9 @@ function get_controls(gradgen::GradGenerator)
 end
 
 
+function get_parameters(gradgen::GradGenerator)
+    return get_parameters(gradgen.G)
+end
+
+
 _exp_prop_convert_operator(::GradGenerator) = Matrix{ComplexF64}
diff --git a/test/Project.toml b/test/Project.toml
index abe5251..e33d663 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -2,6 +2,7 @@
 Coverage = "a2441757-f6aa-5fb2-8edb-039e3f45d037"
 Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
 DocumenterInterLinks = "d12716ef-a0f6-4df4-a9f1-a5a34e75c656"
+GRAPE = "6b52fcaf-80fe-489a-93e9-9f92080510be"
 JuliaFormatter = "98e50ef6-434e-11e9-1051-2b60c6c9e899"
 LiveServer = "16fef848-5104-11e9-1b77-fb7a48bbb589"
 LocalCoverage = "5f6e1e16-694c-5876-87ef-16b5274f298e"