.

Dockerfile

@@ -33,7 +33,7 @@ RUN python3 -m pip install numba==0.58.1
 # RUN python3 -m pip install --no-cache-dir torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
 RUN python3 -m pip install torch torchvision torchaudio
 RUN python3 -m pip install torch_geometric
-RUN python3 -m pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv
+RUN python3 -m pip install  torch_scatter torch_sparse torch_cluster torch_spline_conv
 RUN python3 -m pip install pytorch-lightning yacs torchmetrics
 RUN python3 -m pip install performer-pytorch
 RUN python3 -m pip install tensorboardX