From 96eb123004e4c5bf00edc56c5a2dcfd579983618 Mon Sep 17 00:00:00 2001 From: cont-m-nakagawa Date: Thu, 19 May 2022 17:37:13 +0900 Subject: [PATCH] Correct the wrong type hints --- tensorflow_data_validation/utils/stats_gen_lib.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow_data_validation/utils/stats_gen_lib.py b/tensorflow_data_validation/utils/stats_gen_lib.py index af69c254..85730eab 100644 --- a/tensorflow_data_validation/utils/stats_gen_lib.py +++ b/tensorflow_data_validation/utils/stats_gen_lib.py @@ -51,7 +51,7 @@ def generate_statistics_from_tfrecord( data_location: Text, - output_path: Optional[bytes] = None, + output_path: Optional[str] = None, stats_options: options.StatsOptions = options.StatsOptions(), pipeline_options: Optional[PipelineOptions] = None, ) -> statistics_pb2.DatasetFeatureStatisticsList: @@ -111,7 +111,7 @@ def generate_statistics_from_csv( data_location: Text, column_names: Optional[List[types.FeatureName]] = None, delimiter: Text = ',', - output_path: Optional[bytes] = None, + output_path: Optional[str] = None, stats_options: options.StatsOptions = options.StatsOptions(), pipeline_options: Optional[PipelineOptions] = None, compression_type: Text = CompressionTypes.AUTO,