babyyoda.yoda requires yoda import

src/babyyoda/yoda/histo1d.py

@@ -1,3 +1,4 @@
+import yoda
 from packaging import version
 
 import babyyoda
@@ -9,7 +10,6 @@ def __init__(self, *args, **kwargs):
         """
         target is either a yoda or grogu HISTO1D_V2
         """
-        import yoda
 
         target = args[0] if len(args) == 1 else yoda.Histo1D(*args, **kwargs)
         # unwrap target

src/babyyoda/yoda/histo2d.py

@@ -1,3 +1,4 @@
+import yoda
 from packaging import version
 
 import babyyoda
@@ -9,7 +10,6 @@ def __init__(self, *args, **kwargs):
         """
         target is either a yoda or grogu HISTO2D_V2
         """
-        import yoda
 
         target = args[0] if len(args) == 1 else yoda.Histo2D(*args, **kwargs)
 

src/babyyoda/yoda/read.py

@@ -1,3 +1,5 @@
+import yoda as yd
+
 from babyyoda.yoda.histo1d import Histo1D
 from babyyoda.yoda.histo2d import Histo2D
 
@@ -6,7 +8,6 @@ def read(file_path: str):
     """
     Wrap yoda histograms in the by HISTO1D_V2 class
     """
-    import yoda as yd
 
     ret = {}
     for k, v in yd.read(file_path).items():

src/babyyoda/yoda/write.py

@@ -1,9 +1,9 @@
 import warnings
 
+import yoda as yd
 
-def write(anyhistograms, file_path: str, *args, gz=False, **kwargs):
-    import yoda as yd
 
+def write(anyhistograms, file_path: str, *args, gz=False, **kwargs):
     if gz and not file_path.endswith((".gz", ".gzip")):
         warnings.warn(
             "gz is True but file_path does not end with .gz or .gzip", stacklevel=2