Create unit tests for PytorchWildlife

[aa-hernandez]

Search before asking

• I have searched the Pytorch-Wildlife issues and found no similar bug report.

Description

Help us create unit tests for the different Pytorch-Wildlife modules!

Use case

No response

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!