Update setup_blender.sh

setup_blender.sh

@@ -23,10 +23,14 @@ TORCH_INDEX_URL="https://download.pytorch.org/whl/${TORCH_CUDA_SUFFIX}" # Make s
 PYG_TORCH_VERSION_SUFFIX="2.3.1" # PyTorch version suffix in PyG URLs
 PYG_PYTHON_VERSION="cp311"       # Python version suffix
 PYG_CUDA_SUFFIX="cu121"          # CUDA version suffix
+# Base URL for the directory containing the wheels
 PYG_BASE_URL="https://data.pyg.org/whl/torch-${PYG_TORCH_VERSION_SUFFIX}+${PYG_CUDA_SUFFIX}"
-# Find exact wheel names from the HTML pages if these links break
-TORCH_SCATTER_WHL_URL="${PYG_BASE_URL}/torch_scatter-2.1.2+pt${PYG_TORCH_VERSION_SUFFIX/./}${PYG_CUDA_SUFFIX}-${PYG_PYTHON_VERSION}-${PYG_PYTHON_VERSION}-linux_x86_64.whl"
-TORCH_CLUSTER_WHL_URL="${PYG_BASE_URL}/torch_cluster-1.6.3+pt${PYG_TORCH_VERSION_SUFFIX/./}${PYG_CUDA_SUFFIX}-${PYG_PYTHON_VERSION}-${PYG_PYTHON_VERSION}-linux_x86_64.whl"
+# Correct wheel filenames (checked against the index page)
+TORCH_SCATTER_WHL_FILENAME="torch_scatter-2.1.2-${PYG_PYTHON_VERSION}-${PYG_PYTHON_VERSION}-linux_x86_64.whl"
+TORCH_CLUSTER_WHL_FILENAME="torch_cluster-1.6.3-${PYG_PYTHON_VERSION}-${PYG_PYTHON_VERSION}-linux_x86_64.whl"
+# Construct the full URLs
+TORCH_SCATTER_WHL_URL="${PYG_BASE_URL}/${TORCH_SCATTER_WHL_FILENAME}"
+TORCH_CLUSTER_WHL_URL="${PYG_BASE_URL}/${TORCH_CLUSTER_WHL_FILENAME}"
 
 
 # --- Download and Extract Blender ---
@@ -75,8 +79,8 @@ echo "PyTorch and Torchvision installation complete."
 # --- Install torch-scatter and torch-cluster using direct wheel URLs SECOND ---
 # This avoids build issues by using pre-compiled wheels directly.
 echo "Installing torch-scatter and torch-cluster from specific wheels..."
-echo "Scatter wheel: ${TORCH_SCATTER_WHL_URL}"
-echo "Cluster wheel: ${TORCH_CLUSTER_WHL_URL}"
+echo "Scatter wheel URL: ${TORCH_SCATTER_WHL_URL}"
+echo "Cluster wheel URL: ${TORCH_CLUSTER_WHL_URL}"
 "${BLENDER_PY_EXEC}" -m pip install --no-cache-dir \
     "${TORCH_SCATTER_WHL_URL}" \
     "${TORCH_CLUSTER_WHL_URL}"