CUDA problem

[JessedeDoes]

I get this error:

Traceback (most recent call last):
File "test.py", line 4, in
plot_example(True, reference, translation)
File "/home/jesse/.local/lib/python3.7/site-packages/moverscore_v2.py", line 194, in plot_example
distance_matrix = masks * distance_matrix
RuntimeError: expected backend CUDA and dtype Float but got backend CUDA and dtype Long

(The line in question is
distance_matrix = masks * distance_matrix,

when i change it to distance_matrix = masks.double() * distance_matrix.double(),

the problem seems to disappear)