🐛 Fix module default

Signed-off-by: gkumbhat <[email protected]>
flaviabeo · Nov 6, 2023 · d1e3cba · d1e3cba
1 parent 2ec38e6
commit d1e3cba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/caikit_nlp/toolkit/trainer_utils.py b/caikit_nlp/toolkit/trainer_utils.py
@@ -34,7 +34,7 @@ def validate_training_data(train_stream: DataStream, model_name: str, module_id:
  module_default = (
  get_config()
  .training_data_limit.get(module_id, {})
- .get(module_id, global_default)
+ .get("__default__", global_default)
  )
 
  max_num_examples = (