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