Skip to content

Commit

Permalink
Added collect_summary to usage + minor reformatting
Browse files Browse the repository at this point in the history
  • Loading branch information
w4nderlust committed Sep 22, 2020
1 parent 4c183ee commit 15a744d
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions ludwig/export.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ def export_neuropod(
logger.info('Saved to: {0}'.format(output_path))


def export_savedmodel_cli(sys_argv):
def cli_export_savedmodel(sys_argv):
parser = argparse.ArgumentParser(
description='This script loads a pretrained model '
'and saves it as a SavedModel.',
Expand Down Expand Up @@ -131,7 +131,7 @@ def export_savedmodel_cli(sys_argv):
export_savedmodel(**vars(args))


def export_neuropod_cli(sys_argv):
def cli_export_neuropod(sys_argv):
parser = argparse.ArgumentParser(
description='This script loads a pretrained model '
'and saves it as a Neuropod.',
Expand Down Expand Up @@ -195,10 +195,10 @@ def export_neuropod_cli(sys_argv):
if len(sys.argv) > 1:
if sys.argv[1] == 'savedmodel':
contrib_command("export_savedmodel", *sys.argv)
export_savedmodel_cli(sys.argv[2:])
cli_export_savedmodel(sys.argv[2:])
elif sys.argv[1] == 'neuropod':
contrib_command("export_neuropod", *sys.argv)
export_neuropod_cli(sys.argv[2:])
cli_export_neuropod(sys.argv[2:])
else:
print('Unrecognized command')
else:
Expand Down

0 comments on commit 15a744d

Please sign in to comment.