Who's there?
The spiritual successor to knockknock for PyTorch Lightning, to get a notification when your training is complete or when it crashes during the process with a single callback.
🚀
Features
- Supports E-Mail, Discord, Slack, Teams, Telegram
🎯
Installation
pip install -U whos-there
or install with Poetry
poetry add whos-there
🤯
How to use it
from whos_there.callback import NotificationCallback
from whos_there.senders.debug import DebugSender
trainer = pl.Trainer(
callbacks=[
NotificationCallback(senders=[
# Add your senders here
DebugSender(),
])
]
)
Requires your e-mail provider specific SMTP settings.
from whos_there.senders.email import EmailSender
# ...
EmailSender(
host="smtp.example.de",
port=587,
sender_email="[email protected]",
password="*********",
recipient_emails=[
"[email protected]",
"[email protected]",
]
)
Discord
Requires your Discord channel's webhook URL.
from whos_there.senders.discord import DiscordSender
# ...
DiscordSender(
webhook_url="https://discord.com/api/webhooks/XXXXXXXXXXXXXX/XXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"
)
Slack
Requires your Slack room webhook URL and optionally your user id (if you want to tag yourself or someone else).
from whos_there.senders.slack import SlackSender
# ...
SlackSender(
webhook_url="https://hooks.slack.com/services/T00000000/B00000000/XXXXXXXXXXXXXXXXXXXXXXXX",
channel="channel_name",
user_mentions=[
"XXXXXXXX"
]
)
Teams
Requires your Team Channel webhook URL.
from whos_there.senders.teams import TeamsSender
# ...
TeamsSender(
webhook_url="https://XXXXX.webhook.office.com/",
user_mentions=[
"twsl"
]
)
Telegram
You can also use Telegram Messenger to get notifications. You'll first have to create your own notification bot by following the three steps provided by Telegram here and save your API access TOKEN. Telegram bots are shy and can't send the first message so you'll have to do the first step. By sending the first message, you'll be able to get the chat_id required (identification of your messaging room) by visiting https://api.telegram.org/bot
and get the int under the key message['chat']['id'].
from whos_there.senders.telegram import TelegramSender
# ...
TelegramSender(
chat_id=1234567890,
token="XXXXXXX:XXXXXXXXXXXXXXXXXXXXXXXXXXX"
)
🛡
License
This project is licensed under the terms of the MIT license. See LICENSE for more details.
🏅
Credits
This project was generated with
Big thanks to knockknock for the idea and code snippets.
127 Dec 23, 2022
2.3k Dec 17, 2022
9 May 11, 2022
18 Dec 24, 2022
2 Feb 04, 2022
6 Jan 12, 2022
3 Jan 04, 2022
2 Dec 19, 2022
107 Dec 27, 2022
42 Dec 08, 2022
10 Nov 25, 2022
89 Dec 18, 2022
365 Jan 04, 2023
21.8k Jan 09, 2023
165 Dec 26, 2022
191 Dec 26, 2022
36 Dec 12, 2022
213 Jan 02, 2023
42 Dec 07, 2022
194 Dec 06, 2022