Skip to content
10 changes: 9 additions & 1 deletion butler.py
Original file line number Diff line number Diff line change
Expand Up @@ -236,6 +236,14 @@ def _add_weights_subparser(toplevel_subparsers):
_add_weights_batches_subparser(subparsers)
_add_weights_target_subparser(subparsers)

def _add_reproduce_subparser(toplevel_subparsers):
"""Adds a parser for the `reproduce` command."""
parser = toplevel_subparsers.add_parser(
'reproduce', help='Reproduce a testcase locally.')
parser.add_argument(
'-c', '--config-dir', required=True, help='Path to application config.')
parser.add_argument(
'-t', '--testcase-id', required=True, help='The testcase ID to reproduce.')

def main():
"""Parse the command-line args and invoke the right command."""
Expand Down Expand Up @@ -410,7 +418,7 @@ def main():

_add_integration_tests_subparsers(subparsers)
_add_weights_subparser(subparsers)

_add_reproduce_subparser(subparsers)
args = parser.parse_args()
if not args.command:
parser.print_help()
Expand Down
Loading
Loading