docs: Fix features in adding-a-parameter listing.

pywr · Aug 6, 2024 · 987f29b · 987f29b
1 parent 5aafe40
commit 987f29b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/pywr-book/listings/adding-a-parameter/Cargo.toml b/pywr-book/listings/adding-a-parameter/Cargo.toml
@@ -9,3 +9,7 @@ pywr-schema = { path = "../../../pywr-schema" }
 pywr-schema-macros = { path = "../../../pywr-schema-macros" }
 serde = { workspace = true }
 schemars = { workspace = true }
+
+[features]
+core = ["pywr-schema/core"]
+default = ["core"]
diff --git a/pywr-book/listings/adding-a-parameter/src/main.rs b/pywr-book/listings/adding-a-parameter/src/main.rs
@@ -55,8 +55,12 @@ impl GeneralParameter<f64> for MaxParameter {
 
 // ANCHOR_END: impl-parameter
 mod schema {
+    #[cfg(feature = "core")]
+    use pywr_core::parameters::ParameterIndex;
     use pywr_schema::metric::Metric;
     use pywr_schema::parameters::ParameterMeta;
+    #[cfg(feature = "core")]
+    use pywr_schema::{model::LoadArgs, SchemaError};
     use schemars::JsonSchema;
 
     // ANCHOR: schema