未验证 提交 56eac382 编写于 作者: J Jeff Rasley 提交者: GitHub

fix pytest issue (#1764)

上级 97f8a9eb
......@@ -18,7 +18,8 @@ try:
_amp_available = True
except ImportError:
_amp_available = False
amp_available = pytest.mark.skip(_amp_available, reason="apex/amp is not installed")
amp_available = pytest.mark.skipif(not _amp_available,
reason="apex/amp is not installed")
def test_lamb_fp32_grad_clip(tmpdir):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册