fix: add missing torch import

#43

Included torch import which was previously missing.

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment