Conservative cleanup: preserve numpy/torch tests (public APIs depend on them), only remove scipy/pandas/onnx tests

Author: alexnorell

docker/dockerfiles/Dockerfile.onnx.jetson.6.2.0

@@ -199,14 +199,15 @@ RUN ln -sf /usr/bin/python3 /usr/bin/python && \
         dist/inference_sdk*.whl \
         "setuptools<=75.5.0"
 
-# Remove test directories, development tools, and unnecessary packages to reduce image size (~500MB savings)
+# 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
 RUN cd /usr/local/lib/python3.10/dist-packages && \
-    find . -type d -name tests -exec rm -rf {} + 2>/dev/null || true && \
     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 torch/test || 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