ModuleNotFoundError: No module named 'SuperGluePretrainedNetwork' from official ghcr.io/nerfstudio-project/nerfstudio:latest

[huster-wgm]

Describe the bug

ModuleNotFoundError: No module named 'SuperGluePretrainedNetwork' when using the official ghcr.io/nerfstudio-project/nerfstudio:latest Docker container with HLoc extractors (superpoint.py).

To Reproduce

1. Enter docker container

docker run
--name nerfstudio
--gpus all
--privileged
--shm-size=16g
-v /media/go-hiroaki/WORK/Robotics/SLAM/:/workspace/
-p 7007:7007
-it
ghcr.io/nerfstudio-project/nerfstudio:latest bash

2. Run data preproces pipeline

ns-process-data video --sfm-tool hloc --num-downscales 1 --data ./Datasets/DJI/videos/video_20250727_171450.mp4 --output-dir ./Datasets/nerfstudio/video_20250727_171450

3. Error

File "/usr/local/lib/python3.10/dist-packages/hloc/extractors/superpoint.py", line 8, in
from SuperGluePretrainedNetwork.models import superpoint # noqa E402
ModuleNotFoundError: No module named 'SuperGluePretrainedNetwork'

Screenshots

Additional context

OS: Ubuntu 22.04 (inside container)
Container: ghcr.io/nerfstudio-project/nerfstudio:latest
Python: 3.10+

[huster-wgm]

Temporary Solution

Within container

1. Install git / wget

apt update && apt install git wget

2. Clone missing repo

mkdir -p /usr/local/lib/python3.10/dist-packages/third_party
cd /usr/local/lib/python3.10/dist-packages/third_party && git clone https://github.com/magicleap/SuperGluePretrainedNetwork.git

[hrh-bbc-rd]

I've also experiencedthis error, thank you for the fix.

[astaroph]

Just adding to this in case its helpful for anyone else:
I am using nerfstudio in a distributed environment with Python version 3.10 and to get the fix to work I had to git clone SuperGluePretrainedNetwork.git directly into ENV_NAME/lib/python3.10/site-packages/ instead of into a 'third_party' subdir.