[BUG]No module named 'hierarchical_parameter_server' while I run docker with nvcr.io/nvidia/merlin/merlin-hugectr:24.06 image

[HuuY]

Describe the bug
In order to run the demo fellow this guide:
https://github.com/NVIDIA-Merlin/HugeCTR/blob/main/hps_tf/notebooks/hierarchical_parameter_server_demo.ipynb
I run docker with image nvcr.io/nvidia/merlin/merlin-hugectr:24.06 with:
$ docker run --gpus=all --name huge_ctr_hugo -it --cap-add SYS_NICE -v /home/:/home/ -w /home/hugo -it --privileged nvcr.io/nvidia/merlin/merlin-hugectr:24.06
But when I run $ python3 -c "import hierarchical_parameter_server as hps"
error occurs like No module named 'hierarchical_parameter_server'

So is there something I did wrong or the api changes？
Thanks in advance.

Environment (please complete the following information):

• OS: [e.g. Ubuntu 20.04]
• Graphic card: [e.g. 8 cards nvidia L20]
• CUDA version: [e.g. CUDA 12.6]
• Docker image ：nvcr.io/nvidia/merlin/merlin-hugectr:24.06