ModuleNotFoundError: No module named 'dgl.contrib'

[trebbiano]

Hi, I'm having trouble importing deepvelo even after installing a CUDA-specific version of dgl. Error message and version info below. Thanks in advance for looking into this!

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
Cell In[1], line 1
----> 1 import deepvelo as dv

File ~/.local/lib/python3.8/site-packages/deepvelo/__init__.py:1
----> 1 from .train import *

File ~/.local/lib/python3.8/site-packages/deepvelo/train.py:9
      6 from deepvelo.trainer import Trainer
      7 from scvelo import logging as logg
----> 9 import deepvelo.data_loader.data_loaders as module_data
     10 import deepvelo.model.loss as module_loss
     11 import deepvelo.model.metric as module_metric

File ~/.local/lib/python3.8/site-packages/deepvelo/data_loader/data_loaders.py:9
      7 import dgl
      8 import hnswlib
----> 9 from dgl.contrib.sampling import NeighborSampler
     10 from torch.utils.data import Dataset
     11 from sklearn.metrics import pairwise_distances

ModuleNotFoundError: No module named 'dgl.contrib'

dgl==1.0.0+cu113
deepvelo==0.2.4

System info

DISTRIB_DESCRIPTION="Ubuntu 20.04.5 LTS"
5.15.79.1-microsoft-standard-WSL2

[MishalAshraf]

I am having a similar issue, did you find a way to resolve?

[trebbiano]

Not really. This does not seem to be an actively maintained repo, so I switched to other tools.