Let uproot_methods find single class implementations for templates

scikit-hep · May 7, 2020 · ee56b7f · ee56b7f
1 parent 3c0b015
commit ee56b7f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/uproot_methods/classes/__init__.py b/uproot_methods/classes/__init__.py
@@ -3,8 +3,14 @@
 # BSD 3-Clause License; see https://github.com/scikit-hep/uproot-methods/blob/master/LICENSE
 
 def hasmethods(name):
-    if name not in globals() and name in hasmethods.loaders:
-        globals()[name] = hasmethods.loaders[name].load_module(name)
+    if name not in globals():
+        if name in hasmethods.loaders:
+            globals()[name] = hasmethods.loaders[name].load_module(name)
+        elif '_3c_' in name and '_3e_' in name:
+            bare_name = name.split('_3c_')[0]
+            if bare_name in hasmethods.loaders:
+                globals()[name] = hasmethods.loaders[bare_name].load_module(bare_name)
+
     return name in globals() and isinstance(getattr(globals()[name], "Methods", None), type)
 
 import pkgutil