From faf0a9af6f4c0962639385637464c130cb0c744f Mon Sep 17 00:00:00 2001
From: Lindsey Gray <lindsey.gray@gmail.com>
Date: Wed, 29 Jan 2025 10:25:52 -0600
Subject: [PATCH] add reproducer as test

---
 tests/samples/small_lumi.csv |  4 ++++
 tests/test_lumi_tools.py     | 19 +++++++++++++++++++
 2 files changed, 23 insertions(+)
 create mode 100644 tests/samples/small_lumi.csv

diff --git a/tests/samples/small_lumi.csv b/tests/samples/small_lumi.csv
new file mode 100644
index 000000000..3b6a7a2f0
--- /dev/null
+++ b/tests/samples/small_lumi.csv
@@ -0,0 +1,4 @@
+#Data tag : 24v2 , Norm tag: None
+#run:fill,ls,time,beamstatus,E(GeV),delivered(/pb),recorded(/pb),avgpu,source
+370790:9073,54:54,07/16/23 22:51:10,STABLE BEAMS,6800,0.265372231,0.005674056,33.0,AVG
+368229:8853,74:74,05/31/23 00:23:51,STABLE BEAMS,6800,0.106034314,0.101855712,27.4,AVG
diff --git a/tests/test_lumi_tools.py b/tests/test_lumi_tools.py
index eed65d845..9ad47a731 100644
--- a/tests/test_lumi_tools.py
+++ b/tests/test_lumi_tools.py
@@ -175,3 +175,22 @@ def test_lumilist_client_fromfile():
         (result,) = dask.compute(lumilist.array)
 
         assert result.to_list() == [[1, 13889]]
+
+
+def test_1259_avoid_pickle_numba_dict():
+
+    runs_eager = ak.Array([368229, 368229, 368229, 368229])
+    runs = dak.from_awkward(runs_eager, 2)
+    lumis_eager = ak.Array([74, 74, 74, 74])
+    lumis = dak.from_awkward(lumis_eager, 2)
+
+    def count_lumi(runs, lumis):
+        total_lumi = 0
+        my_lumilist = LumiList(runs, lumis)
+        my_lumidata = LumiData("tests/samples/small_lumi.csv")
+        total_lumi += my_lumidata.get_lumi(my_lumilist)
+        return total_lumi
+
+    with Client() as _:
+        output = count_lumi(runs, lumis)
+        dask.compute(output)[0]