Move ScheduleDag

metno · Apr 29, 2024 · c0d6b46 · c0d6b46
1 parent 0b85c96
commit c0d6b46
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -189,6 +189,14 @@ pub mod dev_utils {
         }
     }
 
+    /// A wrapper struct that separates time series and spatial tests
+    /// into two different DAGs
+    #[derive(Debug, Clone)]
+    pub struct ScheduleDag {
+        pub series: Option<Dag<&'static str>>,
+        pub spatial: Option<Dag<&'static str>>,
+    }
+
     pub fn construct_fake_dag() -> ScheduleDag {
         let mut dag: Dag<&'static str> = Dag::new();
 
@@ -208,12 +216,6 @@ pub mod dev_utils {
         }
     }
 
-    #[derive(Debug, Clone)]
-    pub struct ScheduleDag {
-        pub series: Option<Dag<&'static str>>,
-        pub spatial: Option<Dag<&'static str>>,
-    }
-
     pub fn construct_hardcoded_dag() -> ScheduleDag {
         let mut series: Dag<&'static str> = Dag::new();
         series.add_node("dip_check");