Serialize Weights + PortOffset (#36)

CQCL · May 9, 2023 · 071fcbf · 071fcbf
1 parent 31040fc
commit 071fcbf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -259,6 +259,7 @@ pub struct IndexError {
 
 /// Port offset in a node
 #[derive(Clone, Copy, PartialEq, PartialOrd, Eq, Ord, Hash)]
+#[cfg_attr(feature = "serde", derive(Deserialize, Serialize))]
 pub enum PortOffset {
     /// Input to a node
     ///

diff --git a/src/weights.rs b/src/weights.rs
@@ -40,11 +40,15 @@
 
 use std::ops::{Index, IndexMut};
 
+#[cfg(feature = "serde")]
+use serde::{Deserialize, Serialize};
+
 use crate::{NodeIndex, PortIndex, SecondaryMap};
 
 /// Graph component that encodes node and port weights.
 /// Based on two [`SecondaryMap`] containers.
 #[derive(Debug, Clone)]
+#[cfg_attr(feature = "serde", derive(Deserialize, Serialize))]
 pub struct Weights<N, P> {
     /// Node weights.
     pub nodes: SecondaryMap<NodeIndex, N>,