Describe the bug
When I run 'pytest test/test_tensorclass.py::test_torchsnapshot -v ' and ensure pytorchsnapshot is installed, an assertionerror will occur.
To Reproduce
pip install torchsnapshot
pytest test/test_tensorclass.py::test_torchsnapshot -v
Expected behavior
pass the UT.
Screenshots
System info
Describe the characteristic of your environment:
- Describe how the library was installed (pip, source, ...)
- Python version 3.10.12
- torch 2.5.1 torch 2.7.0
Reason and Possible fixes
Maybe it is a wrong use of torchsnapshot or wrong assert condition.
Checklist