From f568d8d19ff3207ab4a32e937347cbe969140d70 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 14 Sep 2022 16:01:22 +0100 Subject: [PATCH] =?UTF-8?q?io.open=20=E2=86=92=20handle=5Fopen?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit this was we get transparent .gz support --- aimodel/src/subcommands/pretrain_predict.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/pretrain_predict.py b/aimodel/src/subcommands/pretrain_predict.py index 258ee76..060615f 100644 --- a/aimodel/src/subcommands/pretrain_predict.py +++ b/aimodel/src/subcommands/pretrain_predict.py @@ -8,6 +8,7 @@ import re from loguru import logger import tensorflow as tf import numpy as np +from aimodel.src.lib.io.handle_open import handle_open from lib.ai.RainfallWaterContraster import RainfallWaterContraster from lib.dataset.dataset import dataset_predict @@ -64,7 +65,7 @@ def run(args): handle = sys.stdout if filepath_output != "-": - handle = io.open(filepath_output, "w") + handle = handle_open(filepath_output, "w") i = 0 for rainfall, water in ai.embed(dataset):