Merge pull request #1293 from lgray/fix_pfnano

fix: missing momentum class for PFCands

Author: lgray

Diff for: src/coffea/nanoevents/methods/nanoaod.py

@@ -842,6 +842,10 @@ class PFCand(candidate.PtEtaPhiMCandidate, base.NanoCollection):
 
 _set_repr_name("PFCand")
 
+PFCandArray.ProjectionClass2D = vector.TwoVectorArray  # noqa: F821
+PFCandArray.ProjectionClass3D = vector.ThreeVectorArray  # noqa: F821
+PFCandArray.ProjectionClass4D = PFCandArray  # noqa: F821
+PFCandArray.MomentumClass = vector.LorentzVectorArray  # noqa: F821
 
 __all__ = [
     "PtEtaPhiMCollection",