sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-12-22 06:05:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

dataset_mono: address ruff linting errors

Browse source
This commit is contained in:
Starbeamrainbowlabs 2024-12-19 15:22:59 +00:00
parent edfc2721a8
commit 0260e626db
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
aimodel/src/lib/dataset/dataset_mono.py
View file

@ -1,6 +1,5 @@
import os
import math
import json
from loguru import logger
@ -8,7 +7,6 @@ import tensorflow as tf
from lib.dataset.read_metadata import read_metadata
from ..io.readfile import readfile
from .parse_heightmap import parse_heightmap
from .primitives.shuffle import shuffle
from .primitives.remove_isolated_pixels import remove_isolated_pixels
@ -142,7 +140,7 @@ def make_dataset(filepaths, compression_type="GZIP", parallel_reads_multiplier=1
dataset = dataset.shuffle(shuffle_buffer_size)
dataset = dataset.map(parse_item(heightmap=heightmap, **kwargs), num_parallel_calls=tf.data.AUTOTUNE)
if batch_size != None:
if batch_size is not None:
dataset = dataset.batch(batch_size, drop_remainder=True)
if prefetch:
dataset = dataset.prefetch(0 if "NO_PREFETCH" in os.environ else tf.data.AUTOTUNE)