winglian commited on
Commit
cbbf039
·
unverified ·
1 Parent(s): 851ccb1

verbose failure message (#1694)

Browse files
src/axolotl/prompt_strategies/__init__.py CHANGED
@@ -2,9 +2,12 @@
2
 
3
  import importlib
4
  import inspect
 
5
 
6
  from axolotl.prompt_strategies.user_defined import UserDefinedDatasetConfig
7
 
 
 
8
 
9
  def load(strategy, tokenizer, cfg, ds_cfg):
10
  try:
@@ -22,5 +25,8 @@ def load(strategy, tokenizer, cfg, ds_cfg):
22
  if "ds_cfg" in sig.parameters:
23
  load_kwargs["ds_cfg"] = ds_cfg
24
  return func(tokenizer, cfg, **load_kwargs)
25
- except Exception: # pylint: disable=broad-exception-caught
 
 
 
26
  return None
 
2
 
3
  import importlib
4
  import inspect
5
+ import logging
6
 
7
  from axolotl.prompt_strategies.user_defined import UserDefinedDatasetConfig
8
 
9
+ LOG = logging.getLogger("axolotl.prompt_strategies")
10
+
11
 
12
  def load(strategy, tokenizer, cfg, ds_cfg):
13
  try:
 
25
  if "ds_cfg" in sig.parameters:
26
  load_kwargs["ds_cfg"] = ds_cfg
27
  return func(tokenizer, cfg, **load_kwargs)
28
+ except ModuleNotFoundError:
29
+ return None
30
+ except Exception as exc: # pylint: disable=broad-exception-caught
31
+ LOG.error(f"Failed to load prompt strategy `{strategy}`: {str(exc)}")
32
  return None