PicklingError

[iknoorjobs]

Hi,

I am trying to run the tutorial script to train on the MC dataset (link) without any modifications. However, when I fit the model using -

trainer.fit(train_dataset, val_dataset, eval_interval=1, log_interval=5)

It gives PicklingError -

PicklingError                             Traceback (most recent call last)
[<ipython-input-20-744a780ae3be>](https://localhost:8080/#) in <cell line: 1>()
----> 1 trainer.fit(train_dataset, val_dataset, eval_interval=1, log_interval=5)

5 frames
[/usr/local/lib/python3.10/dist-packages/lambeq/training/pytorch_model.py](https://localhost:8080/#) in get_diagram_output(self, diagrams)
    132 
    133         parameters = {k: v for k, v in zip(self.symbols, self.weights)}
--> 134         diagrams = pickle.loads(pickle.dumps(diagrams))  # deepcopy, but faster
    135         for diagram in diagrams:
    136             for b in diagram.boxes:

PicklingError: Can't pickle <class 'discopy.tensor.Box[float64]'>: attribute lookup Box[float64] on discopy.tensor failed