ModuleNotFoundError: No module named 'mamba_ssm.ops.triton.ssd_combined

[bkffadia]

Hello Thank you for the awesome work.

I installed mamba-ssm and used it in my projects and it worked perfectly fine. when I use hydra in my python script I get this message despite mamba-ssm being installed.

Traceback (most recent call last):
  File "/media/fadia/NVMe/./PC/TrainingPhastAll.py", line 36, in <module>
    from hydra.modules.hydra import Hydra
  File "/media/fadia/NVMe/PC/hydra/__init__.py", line 1, in <module>
    from .modules import Hydra, MatrixMixer
  File "/media/fadia/NVMe/PC/hydra/modules/__init__.py", line 1, in <module>
    from .hydra import Hydra
  File "/media/fadia/NVMe/PC/hydra/modules/hydra.py", line 16, in <module>
    from mamba_ssm.ops.triton.ssd_combined import mamba_chunk_scan_combined
ModuleNotFoundError: No module named 'mamba_ssm.ops.triton.ssd_combined'

PS : I am running the script using command-line and in a venv because I'm working in linux and the environment is externally-managed (PEP 668). When I run python -c "from mamba_ssm.ops.triton.ssd_combined import mamba_chunk_scan_combined in the terminal it works perfectly fine.

How can I solve this issue ?