diff --git a/examples/auto_compression/amc/amc_args.py b/examples/auto_compression/amc/amc_args.py
index 3d4edd684eeedc64f253a79b62768c054cd385c9..6688080e76b47d5f553230f991aa4f515d7c9d60 100755
--- a/examples/auto_compression/amc/amc_args.py
+++ b/examples/auto_compression/amc/amc_args.py
@@ -1,6 +1,6 @@
-def add_automl_args(argparser, arch_choices=None, enable_pretrained=False):
+def add_automl_args(argparser):
     """
-    Helper function to make it easier to add command-line arguments for AMC to any application.
+    Helper function which defines command-line arguments specific to AMC.
 
     Arguments:
         argparser (argparse.ArgumentParser): Existing parser to which to add the arguments