Issue importing MPI from mpi4py

[DINHQuangDung1999]

This error occurs when I copy image_train.py to the same level of the main directory ./house_diffusion/image_train.py

$ python3.11 image_train.py --dataset rplan --batch_size 32 --set_name train --target_set 8
Traceback (most recent call last):
  File "C:\Users\My PC\house_diffusion\image_train.py", line 7, in <module>
    from house_diffusion import dist_util, logger
  File "C:\Users\My PC\house_diffusion\house_diffusion\dist_util.py", line 10, in <module>
    from mpi4py import MPI
ImportError: DLL load failed while importing MPI: The specified module could not be found.

Update. I solved this issue by installing directly from https://www.microsoft.com/en-us/download/details.aspx?id=105289