diff --git a/src/axolotl/cli/config.py b/src/axolotl/cli/config.py index 82098eb87..94b3b02b7 100644 --- a/src/axolotl/cli/config.py +++ b/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, diff --git a/src/axolotl/telemetry/errors.py b/src/axolotl/telemetry/errors.py index 98acd6a2c..42cd5aaf3 100644 --- a/src/axolotl/telemetry/errors.py +++ b/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, diff --git a/src/axolotl/train.py b/src/axolotl/train.py index d196e0798..79c558f24 100644 --- a/src/axolotl/train.py +++ b/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