linter

scikit-hep · Jul 2, 2024 · 2630830 · 2630830
1 parent 9147242
commit 2630830
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/hepconvert/root_to_parquet.py b/src/hepconvert/root_to_parquet.py
@@ -4,6 +4,7 @@
 
 import awkward as ak
 import uproot
+from numpy import union1d
 
 
 def root_to_parquet(
@@ -261,8 +262,6 @@ def _filter_branches(tree, keep_branches, drop_branches):
         msg = "Can specify either drop_branches or keep_branches, not both."
         raise ValueError(msg) from None
     keys = []
-    from numpy import union1d
-
     if drop_branches:
         if isinstance(drop_branches, str):
             keys = tree.keys(filter_name=drop_branches)