Skip to content

Commit

Permalink
Fixed neuropod cli
Browse files Browse the repository at this point in the history
  • Loading branch information
w4nderlust committed May 27, 2020
1 parent 9da5f23 commit 30d26aa
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions ludwig/neuropod.py
Original file line number Diff line number Diff line change
Expand Up @@ -275,7 +275,7 @@ def export_neuropod(
)


def cli(sys_argv):
def cli():
parser = argparse.ArgumentParser(
description='This script exports a Ludwig model in the Neuropod format'
)
Expand Down Expand Up @@ -314,7 +314,7 @@ def cli(sys_argv):
default='ludwig_model'
)

args = parser.parse_args(sys_argv)
args = parser.parse_args()

logging.getLogger('ludwig').setLevel(
logging_level_registry[args.logging_level]
Expand All @@ -325,10 +325,10 @@ def cli(sys_argv):
print_ludwig('Export Neuropod', LUDWIG_VERSION)

export_neuropod(
args.ludwig.model_path, args.neuropod_path, args.nueropod_model_name
args.ludwig_model_path, args.neuropod_path, args.neuropod_model_name
)


if __name__ == '__main__':
# contrib_command("neuropod", *sys.argv)
cli(sys.argv)
cli()

0 comments on commit 30d26aa

Please sign in to comment.