Error for importing LogMemoryMixin, MultithreadingMixin in torch_geometric.loader.mixin

[songsong0425]

Hi, while I generating the custom LinkNeighborLoader, I counted the error with some functions in mixin and utils.

from torch_geometric.data import Data, FeatureStore, GraphStore, HeteroData
from torch_geometric.loader.base import DataLoaderIterator
from torch_geometric.loader.mixin import (
    AffinityMixin,
    LogMemoryMixin,
    MultithreadingMixin,
)
from torch_geometric.loader.utils import (
    filter_custom_hetero_store,
    filter_custom_store,
    filter_data,
    filter_hetero_data,
    get_edge_label_index,
    infer_filter_per_worker,
)
...

ImportError: cannot import name 'LogMemoryMixin' from 'torch_geometric.loader.mixin' (/scratch/r902a02/.conda/envs/PyG_scratch1/lib/python3.9/site-packages/torch_geometric/loader/mixin.py)

Not only LogMemoryMixin, but also MultithreadingMixin and filter_custom_hetero_store showed the same error.
The current version of my PyG is 2.4.0.dev20231204.

import torch_geometric

print(torch_geometric.__version__)
#2.4.0.dev20231204

Should I update the version of PyG? Or is there any bug in the library?

[rusty1s]

You would either need to update PyG (these mixins got added recently), or simply remove their dependency from your custom LinkNeighborLoader.

[songsong0425]

Okay, I'll try to update PyG soon. Thanks a lot!