Merge pull request #846 from CoffeaTeam/tidy-old-code

chore: remove references to old awkward1 classes
scikit-hep · Jun 26, 2023 · 1ce3fbf · 1ce3fbf
2 parents cc4c391 + a6dd1fa
commit 1ce3fbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/coffea/jetmet_tools/CorrectedJetsFactory.py b/src/coffea/jetmet_tools/CorrectedJetsFactory.py
@@ -47,7 +47,7 @@ def rand_gauss(item):
 
     def getfunction(layout, depth, **kwargs):
         if isinstance(layout, awkward.contents.NumpyArray) or not isinstance(
-            layout, (awkward.contents.Content, awkward.partition.PartitionedArray)
+            layout, (awkward.contents.Content,)
         ):
             return awkward.contents.NumpyArray(
                 randomstate.normal(size=len(layout)).astype(numpy.float32)