small update / fix

src/axolotl/cli/config.py

@@ -14,8 +14,8 @@ import yaml
 from transformers.utils import is_torch_bf16_gpu_available
 
 from axolotl.integrations.base import PluginManager
-from axolotl.telemetry import TelemetryManager
 from axolotl.telemetry.errors import send_errors
+from axolotl.telemetry.manager import TelemetryManager
 from axolotl.utils.comet_ import setup_comet_env_vars
 from axolotl.utils.config import (
     normalize_cfg_datasets,

src/axolotl/telemetry/errors.py

@@ -148,7 +148,7 @@ def send_errors(func: Callable) -> Callable:
 
                 # Send error telemetry
                 telemetry_manager.send_event(
-                    event_type=f"{module_path}-error",
+                    event_type=f"{module_path}-errored",
                     properties={
                         "exception": str(exception),
                         "stack_trace": stack_trace,

src/axolotl/train.py

@@ -32,8 +32,9 @@ from axolotl.loaders import (
     load_processor,
     load_tokenizer,
 )
-from axolotl.telemetry import TelemetryManager
 from axolotl.telemetry.errors import send_errors
+from axolotl.telemetry.errors import send_errors
+from axolotl.telemetry.manager import TelemetryManager
 from axolotl.utils.ctx_managers.sequence_parallel import SequenceParallelContextManager
 from axolotl.utils.dict import DictDefault
 from axolotl.utils.distributed import cleanup_distributed