huggingface / knockknock

🚪✊Knock Knock: Get notified when your training ends with only two additional lines of code
MIT License
2.78k stars 233 forks source link

Fix raise import error in desktop #72

Open IncubatorShokuhou opened 2 years ago

IncubatorShokuhou commented 2 years ago

Currently, knockknock will only print the error message instead of raise an error when win10toast is not install on win10 desktop. This PR fixes it so that user will correctly see

ImportError: Error: to use Windows Desktop Notifications, you need to install `win10toast` first. Please run `pip install win10toast==0.9`.

instead of

UnboundLocalError: local variable 'ToastNotifier' referenced before assignment