From 1fa61e4fe8b6b917a411b8401d33f6cee8b930c9 Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Sat, 7 May 2022 17:15:20 +0800
Subject: [PATCH] Stringify torch.__version__ before serializing it.

---
 icefall/env.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/icefall/env.py b/icefall/env.py
index c29cbb078..0192d1c11 100644
--- a/icefall/env.py
+++ b/icefall/env.py
@@ -95,7 +95,7 @@ def get_env_info() -> Dict[str, Any]:
         "k2-git-sha1": k2.version.__git_sha1__,
         "k2-git-date": k2.version.__git_date__,
         "lhotse-version": lhotse.__version__,
-        "torch-version": torch.__version__,
+        "torch-version": str(torch.__version__),
         "torch-cuda-available": torch.cuda.is_available(),
         "torch-cuda-version": torch.version.cuda,
         "python-version": sys.version[:3],