_nplikes is now a private interface, need to move to backend instead

ikrommyd · Mar 15, 2023 · c322a3f · c322a3f
1 parent 8ca4afe
commit c322a3f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/coffea/lookup_tools/lookup_base.py b/coffea/lookup_tools/lookup_base.py
@@ -19,17 +19,17 @@ def __call__(self, *args, **kwargs):
             )
 
         def getfunction(args, **kwargs):
-            if type(args) is not list:
-                args = [args]
+            if not isinstance(args, (list, tuple)):
+                args = (args,)
             if all(
                 isinstance(
                     x, (awkward.contents.NumpyArray, awkward.contents.EmptyArray)
                 )
                 or not isinstance(x, (awkward.contents.Content))
                 for x in args
             ):
-                nplike = awkward.nplikes.nplike_of(*args)
-                if not isinstance(nplike, awkward.nplikes.Numpy):
+                nplike = awkward._nplikes.nplike_of(*args)
+                if not isinstance(nplike, awkward._nplikes.Numpy):
                     raise NotImplementedError(
                         "support for cupy/jax/etc. numpy extensions"
                     )

diff --git a/coffea/nanoevents/mapping/base.py b/coffea/nanoevents/mapping/base.py
@@ -89,7 +89,6 @@ def __getitem__(self, key):
                 handle = self.get_column_handle(
                     self._column_source(uuid, treepath), handle_name
                 )
-                print(handle_name, handle)
                 stack.append(self.extract_column(handle, start, stop))
             elif node.startswith("!"):
                 tname = node[1:]