diff --git a/src/python/graph_builder.rs b/src/python/graph_builder.rs index 02faea9..41d85a2 100644 --- a/src/python/graph_builder.rs +++ b/src/python/graph_builder.rs @@ -2289,7 +2289,7 @@ impl PyMLGraphBuilder { let attributes = OperatorOptions::Slice(MLSliceOptions { label: String::new(), starts, - sizes, + sizes: sizes.iter().map(|&d| MLDimension::Static(d)).collect(), strides: strides_vec, });