io.open → handle_open

this was we get transparent .gz support
This commit is contained in:
Starbeamrainbowlabs 2022-09-14 16:01:22 +01:00
parent 9b25186541
commit f568d8d19f
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -8,6 +8,7 @@ import re
from loguru import logger from loguru import logger
import tensorflow as tf import tensorflow as tf
import numpy as np import numpy as np
from aimodel.src.lib.io.handle_open import handle_open
from lib.ai.RainfallWaterContraster import RainfallWaterContraster from lib.ai.RainfallWaterContraster import RainfallWaterContraster
from lib.dataset.dataset import dataset_predict from lib.dataset.dataset import dataset_predict
@ -64,7 +65,7 @@ def run(args):
handle = sys.stdout handle = sys.stdout
if filepath_output != "-": if filepath_output != "-":
handle = io.open(filepath_output, "w") handle = handle_open(filepath_output, "w")
i = 0 i = 0
for rainfall, water in ai.embed(dataset): for rainfall, water in ai.embed(dataset):