Serialize const arrays with serde_with

rust-random · Mar 16, 2023 · e8bd2e0 · e8bd2e0
1 parent a16b932
commit e8bd2e0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/rand_distr/Cargo.toml b/rand_distr/Cargo.toml
@@ -27,6 +27,7 @@ serde1 = ["serde", "rand/serde1"]
 rand = { path = "..", version = "0.9.0", default-features = false }
 num-traits = { version = "0.2", default-features = false, features = ["libm"] }
 serde = { version = "1.0.103", features = ["derive"], optional = true }
+serde_with = { version = "1.14.0", optional = true }
 
 [dev-dependencies]
 rand_pcg = { version = "0.4.0", path = "../rand_pcg" }

diff --git a/rand_distr/src/dirichlet.rs b/rand_distr/src/dirichlet.rs
@@ -13,6 +13,8 @@ use num_traits::Float;
 use crate::{Distribution, Exp1, Gamma, Open01, StandardNormal};
 use rand::Rng;
 use core::fmt;
+#[cfg(feature = "serde_with")]
+use serde_with::serde_as;
 
 /// The Dirichlet distribution `Dirichlet(alpha)`.
 ///
@@ -31,8 +33,8 @@ use core::fmt;
 /// println!("{:?} is from a Dirichlet([1.0, 2.0, 3.0]) distribution", samples);
 /// ```
 #[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
+#[cfg_attr(feature = "serde_with", serde_as)]
 #[derive(Clone, Debug, PartialEq)]
-#[cfg_attr(feature = "serde1", derive(serde::Serialize, serde::Deserialize))]
 pub struct Dirichlet<F, const N: usize>
 where
     F: Float,
@@ -41,6 +43,7 @@ where
     Open01: Distribution<F>,
 {
     /// Concentration parameters (alpha)
+    #[cfg_attr(feature = "serde_with", serde_as(as = "[_; N]"))]
     alpha: [F; N],
 }