log error

gizatechxyz · Mar 20, 2024 · 44852bb · 44852bb
1 parent e86856c
commit 44852bb
Show file tree

Hide file tree

Showing 2 changed files with 1,054 additions and 1,050 deletions.
diff --git a/src/operators/ml/tree_ensemble/tree_ensemble_classifier.cairo b/src/operators/ml/tree_ensemble/tree_ensemble_classifier.cairo
@@ -1,3 +1,4 @@
+use core::byte_array::ByteArrayTrait;
 use core::array::ArrayTrait;
 use core::clone::Clone;
 use core::box::BoxTrait;
@@ -356,7 +357,10 @@ impl TreeEnsembleClassifierImpl<
                                 (*classifier.ensemble.atts.nodes_treeids[*index]).into(),
                                 (*classifier.ensemble.atts.nodes_nodeids[*index]).into()
                             );
-                        t_index.append(class_index.get(key).deref());
+                        t_index
+                            .append(
+                                class_index.get(key).deref_or(panic(array!['here is the error']))
+                            );
                     },
                     Option::None => { break; }
                 };