Add test for custom packages

Author: nevillelyh

packages/README.md

@@ -55,6 +55,9 @@ uv build
 # Or vanilla python build if UV fails
 python3 setup.py bdist_wheel
 
+# Minimum test to check that the package installs and imports
+/build/test.sh 'import <pkg>' dist/<pkg>-*.whl
+
 # Copy tarball and wheel to output directory
 cp -r dist/* /dst
 ```

packages/fairseq.sh

@@ -10,4 +10,6 @@ git checkout d81fac8163364561fd6cd9d82b6ee1ba502c3526
 
 uv build
 
+/build/test.sh 'import fairseq' dist/fairseq-*.whl
+
 cp -r dist/* /dst

packages/mid-splatting.sh

@@ -2,20 +2,31 @@
 
 set -euo pipefail
 
+# Torch is needed but not in build requirements
+# shellcheck disable=SC1091
+source /opt/r8/monobase/activate.sh
+
+# Loosely based on
+# https://github.com/pytorch/pytorch/blob/main/.ci/manywheel/build_cuda.sh
+export TORCH_CUDA_ARCH_LIST='5.0;6.0;7.0;7.5;8.0;8.6;9.0'
+
 git clone https://github.com/autonomousvision/mip-splatting.git
 
 cd mip-splatting
 project_dir=$PWD
 
 git checkout dda02ab5ecf45d6edb8c540d9bb65c7e451345a9
 
-export TORCH_CUDA_ARCH_LIST='7.0;7.5;8.0;8.6+PTX'
-
-cd "$project_dir/submodules/diff-gaussian-rasterization"
 # uv build fails due to missing header, potential bug?
+cd "$project_dir/submodules/diff-gaussian-rasterization"
 python3 setup.py bdist_wheel
-cp dist/* /dst
 
 cd "$project_dir/submodules/simple-knn"
 python3 setup.py bdist_wheel
+
+/build/test.sh \
+    'import diff_gaussian_rasterization; import simple_knn' \
+    "$project_dir/submodules/diff-gaussian-rasterization/dist/"*.whl \
+    "$project_dir/submodules/simple-knn/dist/"*.whl \
+
 cp dist/* /dst

packages/test.sh

@@ -0,0 +1,26 @@
+#!/bin/bash
+
+# Test wheel files produced by package scripts
+
+if [ $# -lt 2 ]; then
+    echo "Usage: $(basename "$0") <SCRIPT> <WHEEL>..."
+    exit 1
+fi
+
+script="$1"
+shift
+
+# Install wheel files in the user layer
+# This should detect any dependency conflicts
+for whl in "$@"; do
+    echo "$whl" >> /tmp/requirements.txt
+done
+/opt/r8/monobase/run.sh monobase.user --requirements /tmp/requirements.txt
+
+# Activate the environment including Cog, monobase, and user venv
+# shellcheck disable=SC1091
+source /opt/r8/monobase/activate.sh
+
+# Run the test script
+python3 -c "$script"
+echo "PASS: $script"