From 1520c2295a274e5e84f607ea10b6bcd90f2cbf3a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Zo=C3=AB=20Bilodeau?=
 <70441641+zbilodea@users.noreply.github.com>
Date: Wed, 3 Jul 2024 12:13:30 +0200
Subject: [PATCH] fix: copy_root when selecting size with entry numbers, losing
 100 entries (#103)

* fixed branch slimming

* linter

* Another small fix with copy_root
---
 src/hepconvert/copy_root.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/hepconvert/copy_root.py b/src/hepconvert/copy_root.py
index ecd5bfa..80637c3 100644
--- a/src/hepconvert/copy_root.py
+++ b/src/hepconvert/copy_root.py
@@ -266,7 +266,10 @@ def copy_root(
                     initial_basket_capacity=initial_basket_capacity,
                     resize_factor=resize_factor,
                 )
-
+                try:
+                    of[tree.name].extend(chunk)
+                except AssertionError:
+                    msg = "Are the branch-names correct?"
             else:
                 try:
                     of[tree.name].extend(chunk)