mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-22 09:13:01 +00:00
io.open → handle_open
this was we get transparent .gz support
This commit is contained in:
parent
9b25186541
commit
f568d8d19f
1 changed files with 2 additions and 1 deletions
|
@ -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):
|
||||||
|
|
Loading…
Reference in a new issue