From b4574e67b4e80f018e83fc78c35dfc2e9f8dbab4 Mon Sep 17 00:00:00 2001
From: Markus Tavenrath <matavenrath@nvidia.com>
Date: Mon, 16 Mar 2026 16:42:04 +0100
Subject: [PATCH] Add MLDimension to GraphBuilder MLOperation

---
 src/python/graph_builder.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,
         });