from deepjets import generate
print list(generate(10, w_pt_min=250, w_pt_max=300))

@@ -27,19 +27,46 @@
 ate%0A
-print list(
+%0Afor leading_jet, subjets, constit in 
 gene
@@ -72,16 +72,17 @@
 erate(10
+0
 , w_pt_m
@@ -102,10 +102,68 @@
 max=300)
-)
+:%0A    print leading_jet, subjets.shape%5B0%5D, constit.shape%5B0%5D
 %0A

improve example test script