如何处理 bash 脚本中 python 的断言错误

Question

这个 bash 脚本在 python3.8 中运行良好。 它检查 pytorch 版本。

#!/usr/bin/env bash

if python -c "import torch" &> /dev/null; then
    if python3 -c 'import torch; assert torch.__version__ <(1,11)' > /dev/null
    then
        echo 'pytorch version must >= 1.11.0';
        exit;
    fi
else
    exit;
fi

但是使用 python 3.9 或更高版本会抛出错误：

 Traceback (most recent call last):
  File "<string>", line 1, in <module>
AssertionError

It seems assert in python doesn't fills > /dev/null in bash script any more, how to handle this case so that my bash scirpt works regardless of python version?

（如果它适用于 python 版本 >=3.8 就足够了）