run_joint_pretrain issue on macos

[paolobiancalana]

Hi there,
I apologize if the question may be trivial but I have just started this course of study. I'm Rigger and I'm very passionate.
I'm trying RigNet for the first time on a mac.
All the dependecis installed correctly and after the creation of the data using gen_dataset, I'm now into the Training session.

I'm having this kind of issue lauching the run_joint_pretrain:

`main(parser.parse_args())
Total params: 3.83M
/Users/paolobiancalana/miniconda3/envs/rignet_cuda11/lib/python3.6/site-packages/torch_geometric/deprecation.py:13: UserWarning: 'data.DataLoader' is deprecated, use 'loader.DataLoader' instead
warnings.warn(out)

Epoch: 1 | LR: 0.00050000
Traceback (most recent call last):

File "", line 1, in
main(parser.parse_args())

File "", line 63, in main
train_loss = train(train_loader, model, optimizer, args)

File "", line 93, in train
for data in train_loader:

File "/Users/paolobiancalana/miniconda3/envs/rignet_cuda11/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()

File "/Users/paolobiancalana/miniconda3/envs/rignet_cuda11/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 561, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration

File "/Users/paolobiancalana/miniconda3/envs/rignet_cuda11/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
return self.collate_fn(data)

File "/Users/paolobiancalana/miniconda3/envs/rignet_cuda11/lib/python3.6/site-packages/torch_geometric/loader/dataloader.py", line 20, in call
self.exclude_keys)

File "/Users/paolobiancalana/miniconda3/envs/rignet_cuda11/lib/python3.6/site-packages/torch_geometric/data/batch.py", line 75, in from_data_list
exclude_keys=exclude_keys,

File "/Users/paolobiancalana/miniconda3/envs/rignet_cuda11/lib/python3.6/site-packages/torch_geometric/data/collate.py", line 86, in collate
increment)

File "/Users/paolobiancalana/miniconda3/envs/rignet_cuda11/lib/python3.6/site-packages/torch_geometric/data/collate.py", line 133, in _collate
incs = get_incs(key, values, data_list, stores)

File "/Users/paolobiancalana/miniconda3/envs/rignet_cuda11/lib/python3.6/site-packages/torch_geometric/data/collate.py", line 222, in get_incs
for value, data, store in zip(values, data_list, stores)

File "/Users/paolobiancalana/miniconda3/envs/rignet_cuda11/lib/python3.6/site-packages/torch_geometric/data/collate.py", line 222, in
for value, data, store in zip(values, data_list, stores)

TypeError: inc() takes 3 positional arguments but 4 were given`

can you please help me because I'm stuck and also if you want, explain me why I would be grateful.

Thank you in advance,
Paolo