dev

qubole · Jul 17, 2020 · 425fee7 · 425fee7
1 parent e3dd7db
commit 425fee7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/qds_sdk/clusterv2.py b/qds_sdk/clusterv2.py
@@ -148,7 +148,7 @@ def set_cluster_info_from_arguments(self, arguments):
                               enable_ganglia_monitoring=arguments.enable_ganglia_monitoring,
                               datadog_api_token=arguments.datadog_api_token,
                               datadog_app_token=arguments.datadog_app_token,
-                              notification_channels=arguments.notifications,
+                              notification_channels=arguments.notification_channels,
                               node_bootstrap=arguments.node_bootstrap_file,
                               master_instance_type=arguments.master_instance_type,
                               slave_instance_type=arguments.slave_instance_type,
@@ -710,7 +710,7 @@ def cluster_info_parser(argparser, action):
                                    default=None,
                                    help="overrides for airflow cluster", )
         notifications_group = argparser.add_argument_group("notifications")
-        notifications_group.add_argument("--notification_channels",
+        notifications_group.add_argument("--notification-channels",
                                         nargs="*",
                                         type=int,
                                         dest="notification_channels",

diff --git a/tests/test_clusterv2.py b/tests/test_clusterv2.py
@@ -779,7 +779,7 @@ def test_start_stop_timeouts_invalid(self):
 
     def test_notifications_given(self):
         sys.argv = ['qds.py', '--version', 'v2', 'cluster', 'create', '--label', 'test_label',
-                        '--notification_channels', '7']
+                        '--notification-channels', '7']
         print_command()
         Connection._api_call = Mock(return_value={})
         qds.main()