temporary travis check

This commit is contained in:
erogol 2020-11-17 14:17:03 +01:00
parent 79ed5debcd
commit 8b0e0846a3
1 changed files with 4 additions and 4 deletions

View File

@ -13,10 +13,10 @@ def set_amp_context(mixed_precision):
if mixed_precision:
cm = torch.cuda.amp.autocast()
else:
if platform.python_version() <= "3.6.0":
cm = contextlib.suppress()
else:
cm = contextlib.nullcontext()
# if platform.python_version() <= "3.6.0":
cm = contextlib.suppress()
# else:
# cm = contextlib.nullcontext()
return cm