diff --git a/deepconsensus/preprocess/preprocess.py b/deepconsensus/preprocess/preprocess.py index 84d0d67..0e0be37 100644 --- a/deepconsensus/preprocess/preprocess.py +++ b/deepconsensus/preprocess/preprocess.py @@ -36,7 +36,6 @@ --truth_split=truth_split.tsv \ --output=examples-@split.tfrecord.gz \ --cpus=4 - """ import collections @@ -76,7 +75,7 @@ 'The output filename must end in .tfrecord.gz')) flags.DEFINE_string('truth_to_ccs', None, 'Input truth bam aligned to ccs.') flags.DEFINE_string('truth_bed', None, 'Input truth bedfile.') -# TODO TODO +# TODO flags.DEFINE_string('truth_split', None, 'Input file defining train/eval/test splits.') flags.DEFINE_integer( @@ -122,6 +121,7 @@ def wrap(*args, **kwargs): except: # pylint: disable=bare-except logging.exception('Error in function %s.', f.__name__) raise Exception('Error in worker process') + return wrap @@ -311,6 +311,7 @@ def main(unused_argv) -> None: json_summary = json.dumps(summary, indent=True) summary_file.write(json_summary) + if __name__ == '__main__': logging.use_python_logging() app.run(main)