From 483ad41e6949b1f36f349b28be0249680cebc2ec Mon Sep 17 00:00:00 2001 From: rockerBOO Date: Tue, 15 Jul 2025 19:55:44 -0400 Subject: [PATCH] Add import hint --- train_network.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/train_network.py b/train_network.py index 2eabcdf3..18343ce3 100644 --- a/train_network.py +++ b/train_network.py @@ -45,7 +45,11 @@ from library.custom_train_functions import ( apply_masked_loss, ) from library.utils import setup_logging, add_logging_arguments -from wavelet_loss import WaveletLoss + +try: + from wavelet_loss import WaveletLoss +except: + raise ImportError("wavelet-loss is not installed. Install it with `pip install git+https://github.com/rockerBOO/wavelet-loss`") setup_logging() import logging