fix(hansbug): fix problem in unittest

[HansBug]

This is related to the bug of torch: pytorch/pytorch#69894

[codecov]

Codecov Report

Merging #39 (ad94ebc) into main (be6ec45) will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##             main      #39   +/-   ##
=======================================
  Coverage   99.22%   99.22%           
=======================================
  Files          35       35           
  Lines        2071     2071           
=======================================
  Hits         2055     2055           
  Misses         16       16           

Flag	Coverage Δ
unittests	99.22% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update e380285...ad94ebc. Read the comment docs.