Change Telemetry Config

Author: AryanAhadinia

src/swiss_ai_model_launch/cli/configuration/init_wizard.py

@@ -93,37 +93,15 @@ class InitConfig(ChainConfiguration):
                     prompt="What is your CSCS API key? "
                     "(https://serving.swissai.svc.cscs.ch)",
                 ),
-                BranchConfiguration(
-                    name="telemetry_configuration",
-                    head_configuration=OptionsConfiguration(
-                        name="telemetry",
-                        prompt="Do you want to enable telemetry to help us improve "
-                        "the product? (you can change this later in the config file)",
-                        options={
-                            "default": (
-                                "Yes",
-                                "Anonymized data will be sent to telemetry endpoint.",
-                            ),
-                            "disabled": (
-                                "No",
-                                "No data will be recorded out of your usage.",
-                            ),
-                        },
-                    ),
-                    branches={
-                        "default": ChainConfiguration(
-                            name="telemetry_endpoint_configuration",
-                            chain=[
-                                TextConfiguration(
-                                    name="telemetry_endpoint",
-                                    prompt="What is the endpoint to which telemetry "
-                                    "data should be sent?",
-                                    default="https://sml-dev.swissai.svc.cscs.ch/launches",
-                                ),
-                            ],
+                ChainConfiguration(
+                    name="telemetry_endpoint_configuration",
+                    chain=[
+                        TextConfiguration(
+                            name="telemetry_endpoint",
+                            prompt="Where to send telemetry reports?",
+                            default="https://sml-dev.swissai.svc.cscs.ch/launches",
                         ),
-                        "disabled": None,
-                    },
+                    ],
                 ),
             ],
         )

src/swiss_ai_model_launch/cli/main.py

@@ -227,11 +227,7 @@ async def _main() -> None:
     launcher_type = config.get_non_none_value("launcher")
     if launcher_type == "firecrest":
         firecrest_client = _get_firecrest_client_from_init_config(config)
-        telemetry_endpoint = (
-            config.get_value("telemetry_endpoint")
-            if config.get_non_none_value("telemetry") == "default"
-            else None
-        )
+        telemetry_endpoint = config.get_non_none_value("telemetry_endpoint")
         launcher = await _get_firecrest_launcher_with_client(
             firecrest_client,
             telemetry_endpoint=telemetry_endpoint,