address Jim's comments

scikit-hep · Apr 29, 2022 · 1dfaab3 · 1dfaab3
1 parent 1ae5c42
commit 1dfaab3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/awkward/_v2/_connect/rdataframe/to_rdataframe.py b/src/awkward/_v2/_connect/rdataframe/to_rdataframe.py
@@ -10,6 +10,11 @@
 
 compiler = ROOT.gInterpreter.Declare
 compiler_lock = threading.Lock()
+compiler(
+    """
+#include <Python.h>
+"""
+)
 
 
 def compile(source_code):
@@ -144,8 +149,6 @@ def data_frame(self, layouts):
 
         if not hasattr(ROOT, array_data_source):
             cpp_code = f"""
-#include <Python.h>
-
 namespace awkward {{
 
     class {array_data_source} final
@@ -258,5 +261,4 @@ class {array_data_source} final
             (self.data_ptrs_list),
         )
 
-        rdf.lookups = self.lookups
         return rdf