Remove .pyi deletion - lazy_loader (skimage) requires stub files

Author: alexnorell

docker/dockerfiles/Dockerfile.onnx.jetson.6.2.0

@@ -201,16 +201,15 @@ RUN ln -sf /usr/bin/python3 /usr/bin/python && \
 
 # Remove test directories, development tools, and unnecessary packages to reduce image size
 # This happens AFTER all package installations to ensure everything gets cleaned
-# Note: Keep numpy/torch test dirs (public APIs depend on them), remove from other packages
+# Note: Keep numpy/torch test dirs (public APIs depend on them), keep .pyi files (lazy_loader needs them)
 RUN cd /usr/local/lib/python3.10/dist-packages && \
     find . -type d -name __pycache__ -exec rm -rf {} + 2>/dev/null || true && \
     rm -rf debugpy* jupyterlab* jupyter_* notebook* ipython* ipykernel* || true && \
     rm -rf torch/bin torch/include || true && \
     rm -rf onnx/backend/test onnx/test || true && \
     rm -rf scipy/*/tests pandas/tests || true && \
     rm -rf */examples */benchmarks */docs || true && \
-    rm -rf skimage/data || true && \
-    find . -name "*.pyi" -delete 2>/dev/null || true
+    rm -rf skimage/data || true
 
 # Stage 2: Runtime
 FROM nvcr.io/nvidia/l4t-cuda:12.6.11-runtime