log warning when dataset strategy is not loadable

axolotl-ai-cloud · Jan 23, 2024 · 76b5c2d · 76b5c2d
1 parent 29663d8
commit 76b5c2d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/axolotl/prompt_strategies/dpo/__init__.py b/src/axolotl/prompt_strategies/dpo/__init__.py
@@ -3,6 +3,9 @@
 """
 
 import importlib
+import logging
+
+LOG = logging.getLogger("axolotl")
 
 
 def load(strategy, cfg):
@@ -14,4 +17,5 @@ def load(strategy, cfg):
         load_kwargs = {}
         return func(cfg, **load_kwargs)
     except Exception:  # pylint: disable=broad-exception-caught
+        LOG.warning(f"unable to load strategy {strategy}")
         return None