search

rapidsai / raft

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.
https://docs.rapids.ai/api/raft/stable/
Apache License 2.0
680 stars 180 forks source link

[BUG] Error when running raft_ann_bench #2356

Open tarang-jain opened 3 weeks ago

tarang-jain commented 3 weeks ago

Noticed an error after raft-ann-bench pkg is renamed to raft_ann_bench.

Traceback (most recent call last):
  File "<frozen runpy>", line 198, in _run_module_as_main
  File "<frozen runpy>", line 88, in _run_code
  File "/raid/tarangj/miniconda3/envs/all_cuda-122_arch-x86_64/lib/python3.11/site-packages/raft_ann_bench/run/__main__.py", line 614, in <module>
    main()
  File "/raid/tarangj/miniconda3/envs/all_cuda-122_arch-x86_64/lib/python3.11/site-packages/raft_ann_bench/run/__main__.py", line 545, in main
    validator = import_module(module)
                ^^^^^^^^^^^^^^^^^^^^^
  File "/raid/tarangj/miniconda3/envs/all_cuda-122_arch-x86_64/lib/python3.11/importlib/__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "<frozen importlib._bootstrap>", line 1204, in _gcd_import
  File "<frozen importlib._bootstrap>", line 1176, in _find_and_load
  File "<frozen importlib._bootstrap>", line 1126, in _find_and_load_unlocked
  File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed
  File "<frozen importlib._bootstrap>", line 1204, in _gcd_import
  File "<frozen importlib._bootstrap>", line 1176, in _find_and_load
  File "<frozen importlib._bootstrap>", line 1140, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'raft-ann-bench'