From 57ca12e46d4328a8dfbcb64c925c64a9a3e52c72 Mon Sep 17 00:00:00 2001
From: Dmitry Kalinkin <dkalinkin@bnl.gov>
Date: Fri, 7 Jul 2023 02:00:09 +0000
Subject: [PATCH] kinematics_correlations.py: ak.count -> ak.num where needed

---
 benchmarks/dis/analysis/kinematics_correlations.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/benchmarks/dis/analysis/kinematics_correlations.py b/benchmarks/dis/analysis/kinematics_correlations.py
index 3113df0e..950d1ecb 100644
--- a/benchmarks/dis/analysis/kinematics_correlations.py
+++ b/benchmarks/dis/analysis/kinematics_correlations.py
@@ -46,8 +46,8 @@ def Q2correlation(minq2,method): #minq2 can be 1,10,100, or 1000; method can be
     Q2_List_T = Q2values_T #Truth Q2 values, mapped along x axis
     Q2_List_Y = Q2values_Y #method (E/DA/JB) Q2 values, mapped along y axis
 
-    T_len = ak.count(Q2_List_T,axis=0) #total number of events in Truth
-    Y_len = ak.count(Q2_List_Y,axis=0) #total number of events in method
+    T_len = ak.num(Q2_List_T,axis=0) #total number of events in Truth
+    Y_len = ak.num(Q2_List_Y,axis=0) #total number of events in method
 
     if T_len > Y_len: #if total number of events for Truth is greater
         Y_boolean = ak.count(Q2_List_Y,axis=-1) >= 1 #boolean to filter ak.Arrays wrt single events in method 
@@ -103,8 +103,8 @@ def Xcorrelation(minq2,method): #minq2 can be 1,10,100, or 1000; method can be '
     X_List_T = Xvalues_T #Truth x values, mapped along x axis
     X_List_Y = Xvalues_Y #method (E/DA/JB) x values, mapped along y axis
 
-    T_len = ak.count(X_List_T,axis=0) #total number of events in Truth
-    Y_len = ak.count(X_List_Y,axis=0) #total number of events in method
+    T_len = ak.num(X_List_T,axis=0) #total number of events in Truth
+    Y_len = ak.num(X_List_Y,axis=0) #total number of events in method
 
     if T_len > Y_len: #if total number of events for Truth is greater
         Y_boolean = ak.count(X_List_Y,axis=-1) >= 1 #boolean to filter ak.Arrays wrt single events in method 
-- 
GitLab