From ea976b0543c7fa97628c41c4a936e3113896d18a Mon Sep 17 00:00:00 2001 From: erogol Date: Fri, 6 Nov 2020 13:34:11 +0100 Subject: [PATCH] python compat update for contextlib --- TTS/utils/generic_utils.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/TTS/utils/generic_utils.py b/TTS/utils/generic_utils.py index 686a3453..9ed2e20e 100644 --- a/TTS/utils/generic_utils.py +++ b/TTS/utils/generic_utils.py @@ -3,7 +3,7 @@ import glob import os import shutil import subprocess -from contextlib import nullcontext +import contextlib import torch @@ -12,7 +12,10 @@ def set_amp_context(mixed_precision): if mixed_precision: cm = torch.cuda.amp.autocast() else: - cm = nullcontext() + if os.python.version<=3.6: + cm = contextlib.suppress() + else: + cm = nullcontext() return cm