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: if mixed_precision:
cm = torch.cuda.amp.autocast() cm = torch.cuda.amp.autocast()
else: else:
if platform.python_version() <= "3.6.0": # if platform.python_version() <= "3.6.0":
cm = contextlib.suppress() cm = contextlib.suppress()
else: # else:
cm = contextlib.nullcontext() # cm = contextlib.nullcontext()
return cm return cm