fix: Uproot should be able to run without Awkward. (#831)

src/uproot/models/RNTuple.py

@@ -7,7 +7,6 @@
 import struct
 import zlib
 
-import awkward as ak
 import numpy
 
 import uproot
@@ -213,6 +212,8 @@ def page_list_envelopes(self):
         return self._page_list_envelopes
 
     def base_col_form(self, cr, col_id, parameters=None):
+        ak = uproot.extras.awkward()
+
         form_key = f"column-{col_id}"
         dtype_byte = cr.type
         if dtype_byte == uproot.const.rntuple_role_union:
@@ -230,6 +231,8 @@ def base_col_form(self, cr, col_id, parameters=None):
             return form_key
 
     def col_form(self, field_id):
+        ak = uproot.extras.awkward()
+
         # FIXME remove this ugly logic
         rel_crs = []
         rel_crs_idxs = []
@@ -257,6 +260,8 @@ def col_form(self, field_id):
             raise (RuntimeError(f"Missing special case: {field_id}"))
 
     def field_form(self, this_id, seen):
+        ak = uproot.extras.awkward()
+
         field_records = self.header.field_records
         this_record = field_records[this_id]
         seen.append(this_id)
@@ -310,6 +315,8 @@ def field_form(self, this_id, seen):
             raise AssertionError("this should be unreachable")
 
     def to_akform(self):
+        ak = uproot.extras.awkward()
+
         field_records = self.header.field_records
         recordlist = []
         topnames = self.keys()
@@ -382,6 +389,8 @@ def arrays(
         decompression_executor=None,
         array_cache=None,
     ):
+        ak = uproot.extras.awkward()
+
         entry_stop = entry_stop or self._length
 
         clusters = self.cluster_summaries
@@ -431,6 +440,8 @@ def _split_switch_bits(content):
 
 
 def _recursive_find(form, res):
+    ak = uproot.extras.awkward()
+
     if hasattr(form, "form_key"):
         res.append(form.form_key)
     if hasattr(form, "contents"):

tests/test_0630-rntuple-basics.py

@@ -10,6 +10,8 @@
 
 import uproot
 
+pytest.importorskip("awkward")
+
 
 def test_flat():
     filename = skhep_testdata.data_path("test_ntuple_int_float.root")

tests/test_0662-rntuple-stl-containers.py

@@ -4,13 +4,14 @@
 import queue
 import sys
 
-import awkward as ak
 import numpy
 import pytest
 import skhep_testdata
 
 import uproot
 
+ak = pytest.importorskip("awkward")
+
 
 def test_rntuple_stl_containers():
     filename = skhep_testdata.data_path("test_ntuple_stl_containers.root")

tests/test_0705-rntuple-writing-metadata.py

@@ -5,13 +5,14 @@
 import queue
 import sys
 
-import awkward as ak
 import numpy
 import pytest
 import skhep_testdata
 
 import uproot
 
+ak = pytest.importorskip("awkward")
+
 
 def test_header(tmp_path):
     filepath = os.path.join(tmp_path, "test.root")

tests/test_0808-fix_awkward_form_for_AsStridedObjects.py

@@ -6,6 +6,8 @@
 
 import uproot
 
+pytest.importorskip("awkward")
+
 
 def test():
     path = skhep_testdata.data_path("uproot-HZZ-objects.root")

tests/test_0816-separate-AwkwardForth-machines-by-TBranch.py

@@ -6,6 +6,8 @@
 
 import uproot
 
+pytest.importorskip("awkward")
+
 
 @pytest.mark.parametrize("do_hack", [False, True])
 def test(do_hack):