fix torchvision dependency test

[liopeer]

Changes

• changes how we test if the torchvision.transforms.v2 is available

Necessity

• Testing if we can import torchvision.transforms.v2 is not enough since we need to make sure that we install a version that also has a stable version of the API. Trying to mport Mask should be safer.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 85.24%. Comparing base (c25309c) to head (37ebd7c).
Report is 1 commits behind head on master.

Additional details and impacted files

@@           Coverage Diff           @@
##           master    #1779   +/-   ##
=======================================
  Coverage   85.24%   85.24%           
=======================================
  Files         163      163           
  Lines        6832     6832           
=======================================
  Hits         5824     5824           
  Misses       1008     1008           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.