trainwatch-notify 0.1.0

Lightweight training awareness system for machine learning workflows.

TrainWatch

A lightweight training awareness system for machine learning workflows.

TrainWatch notifies developers when ML training completes or fails and provides a minimal run summary. It is designed to be easy to integrate, framework-agnostic, and reliable.

Install

pip install trainwatch-notify

Note: the PyPI distribution name is trainwatch-notify, but the import package remains trainwatch.

Quickstart

from trainwatch import monitor

monitor.start()

try:
    # Your training loop here.
    monitor.log(epoch=1, loss=0.42, val_acc=0.81)
    monitor.end()
except Exception as exc:
    monitor.fail(exc)
    raise

Convenience context manager:

from trainwatch import monitor

with monitor.watch():
    # Your training loop here.
    pass

Configuration

TrainWatch reads configuration from trainwatch_config.yaml by default. You can also set TRAINWATCH_CONFIG to an absolute path or pass a config dict to monitor.end(config=...) and monitor.fail(config=...).

Example config (YAML):

notifications:
  email: true
  telegram: false

email:
  host: smtp.example.com
  port: 587
  username: [e-mail]
  password: app_password
  sender: [e-mail]
  recipient: [e-mail]
  use_tls: true
  subject: TrainWatch Notification

logging:
  enabled: true
  path: trainwatch_run.json

An example config is available at examples/trainwatch_config.yaml.

Cloud Notifications (Resend + Cloudflare)

If you want users to receive email without SMTP setup, use the TrainWatch Cloud backend.

1. Deploy the Cloudflare Worker in cloudflare/ and set RESEND_API_KEY and RESEND_FROM.
2. Set the backend URL locally:

export TRAINWATCH_BASE_URL="https://your-worker.workers.dev"

3. Register an email once:

from trainwatch import add_email

add_email("[e-mail]")

After that, monitor.end() and monitor.fail() will send notifications automatically if credentials exist. Cloud backend enforces a default limit of 10 emails per user per day.

To remove the email:

from trainwatch import delete_email

delete_email()

CLI

You can also register/delete emails via the CLI:

trainwatch add-email [e-mail]
trainwatch delete-email
trainwatch help

Mid-Run Notifications

Manual milestone:

from trainwatch import monitor

monitor.notify("Model loaded on GPU")

Time-based heartbeat:

monitor.heartbeat(interval_seconds=900)  # every 15 minutes
# ...
monitor.stop_heartbeat()

Step-based pings:

for epoch in range(100):
    train()
    monitor.step(notify_every=10)

Configure defaults once:

monitor.configure(
    heartbeat_interval=900,
    step_notify_every=10,
    heartbeat_message="Training still running",
)

Optional per-channel minimum intervals:

limits:
  email_min_interval_seconds: 1800
  telegram_min_interval_seconds: 300
  cloud_min_interval_seconds: 1800

Environment Variables

These environment variables override config values when set.

• TRAINWATCH_CONFIG
• TRAINWATCH_BASE_URL
• TRAINWATCH_API_KEY
• TRAINWATCH_CREDENTIALS_PATH
• TRAINWATCH_DISABLE_PROXY
• TRAINWATCH_NOTIFICATIONS_EMAIL
• TRAINWATCH_NOTIFICATIONS_TELEGRAM
• TRAINWATCH_EMAIL_HOST
• TRAINWATCH_EMAIL_PORT
• TRAINWATCH_EMAIL_USERNAME
• TRAINWATCH_EMAIL_PASSWORD
• TRAINWATCH_EMAIL_SENDER
• TRAINWATCH_EMAIL_RECIPIENT
• TRAINWATCH_EMAIL_USE_TLS
• TRAINWATCH_EMAIL_SUBJECT
• TRAINWATCH_TELEGRAM_BOT_TOKEN
• TRAINWATCH_TELEGRAM_CHAT_ID
• TRAINWATCH_LOGGING_ENABLED
• TRAINWATCH_LOGGING_PATH

Logging

Set logging.enabled: true in the config to write a run summary JSON file. The default path is trainwatch_run.json relative to the config file location.

Testing

Install dev dependencies and run tests:

pip install -e .[dev]
pytest

Vision

TrainWatch aims to become an intelligent training awareness layer that lets developers manage and monitor long-running training processes without constant supervision. The long-term goal is an intelligent training companion that integrates into broader ML tooling ecosystems and improves developer productivity.

Mission

Provide a lightweight, reliable, easy-to-integrate system that captures training lifecycle events and delivers concise summaries through notification channels such as email or messaging services. The mission prioritizes:

• Simplicity of integration
• Reliability of notifications
• Minimal disruption to existing workflows

Problem Statement

Long-running ML training jobs often require repeated manual checks for completion or failure. This interrupts workflow and delays awareness of errors. Existing enterprise platforms are too heavy for many developers, and there is a gap for a simple, developer-friendly tool focused on basic training awareness.

Value Proposition

TrainWatch allows developers to step away during training without losing awareness of outcomes. Minimal integration and reliable notifications reduce monitoring fatigue and improve experimentation efficiency.

License

Licensed under the GNU Lesser General Public License v3.0 (LGPL-3.0). See LICENSE.