fix bug in scheduler

tensorlakeai · Jul 23, 2024 · 53a0552 · 53a0552
1 parent 3e97551
commit 53a0552
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/scheduler/mod.rs b/src/scheduler/mod.rs
@@ -211,8 +211,10 @@ impl Scheduler {
 
         let mut output_mapping: HashMap<String, String> = HashMap::new();
         for name in extractor.outputs.keys() {
-            let table_name = extraction_policy.output_table_mapping.get(name).unwrap();
-            output_mapping.insert(name.clone(), table_name.clone());
+            let table_name = extraction_policy.output_table_mapping.get(name);
+            if let Some(table_name) = table_name {
+                output_mapping.insert(name.clone(), table_name.clone());
+            }
         }
 
         let mut hasher = DefaultHasher::new();