diff --git a/eli5/xgboost.py b/eli5/xgboost.py index f565d9c8..8a3d0f5d 100644 --- a/eli5/xgboost.py +++ b/eli5/xgboost.py @@ -356,18 +356,19 @@ def _parse_tree_dump(text_dump): def _parse_dump_line(line): # type: (str) -> Tuple[int, Dict[str, Any]] branch_match = re.match( - '^(\t*)(\d+):\[([^<]+)<([^\]]+)\] ' + '^(\t*)(\d+):\[([^<]+)