Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix: convert
torch.__version__
to str
when serializing
```py >>> type(torch.__version__) <class 'torch.torch_version.TorchVersion'> ``` This causes a YAML error: ``` File "/home/jz748/anaconda3/lib/python3.10/site-packages/yaml/representer.py", line 231, in represent_undefined raise RepresenterError("cannot represent an object", data) yaml.representer.RepresenterError: ('cannot represent an object', '2.3.1+cu121') ``` Signed-off-by: Jinzhe Zeng <[email protected]>
- Loading branch information