@@ -527,14 +527,18 @@ impl<H: Hamiltonian, A: AdaptStrategy> StatsBuilder<H, A> {
527
527
528
528
let gradient = if settings. store_gradient {
529
529
let items = MutablePrimitiveArray :: new ( ) ;
530
- Some ( MutableFixedSizeListArray :: new_with_field ( items, "item" , false , dim) )
530
+ Some ( MutableFixedSizeListArray :: new_with_field (
531
+ items, "item" , false , dim,
532
+ ) )
531
533
} else {
532
534
None
533
535
} ;
534
536
535
537
let unconstrained = if settings. store_gradient {
536
538
let items = MutablePrimitiveArray :: new ( ) ;
537
- Some ( MutableFixedSizeListArray :: new_with_field ( items, "item" , false , dim) )
539
+ Some ( MutableFixedSizeListArray :: new_with_field (
540
+ items, "item" , false , dim,
541
+ ) )
538
542
} else {
539
543
None
540
544
} ;
@@ -572,9 +576,9 @@ impl<H: Hamiltonian, A: AdaptStrategy> ArrowBuilder<NutsSampleStats<H::Stats, A:
572
576
store
573
577
. try_push (
574
578
value
575
- . gradient ( )
576
- . as_ref ( )
577
- . map ( |vals| vals. iter ( ) . map ( |& x| Some ( x) ) )
579
+ . gradient ( )
580
+ . as_ref ( )
581
+ . map ( |vals| vals. iter ( ) . map ( |& x| Some ( x) ) ) ,
578
582
)
579
583
. unwrap ( ) ;
580
584
}
@@ -583,9 +587,9 @@ impl<H: Hamiltonian, A: AdaptStrategy> ArrowBuilder<NutsSampleStats<H::Stats, A:
583
587
store
584
588
. try_push (
585
589
value
586
- . unconstrained ( )
587
- . as_ref ( )
588
- . map ( |vals| vals. iter ( ) . map ( |& x| Some ( x) ) )
590
+ . unconstrained ( )
591
+ . as_ref ( )
592
+ . map ( |vals| vals. iter ( ) . map ( |& x| Some ( x) ) ) ,
589
593
)
590
594
. unwrap ( ) ;
591
595
}
@@ -634,7 +638,11 @@ impl<H: Hamiltonian, A: AdaptStrategy> ArrowBuilder<NutsSampleStats<H::Stats, A:
634
638
}
635
639
636
640
if let Some ( mut unconstrained) = self . unconstrained . take ( ) {
637
- fields. push ( Field :: new ( "unconstrained_draw" , unconstrained. data_type ( ) . clone ( ) , true ) ) ;
641
+ fields. push ( Field :: new (
642
+ "unconstrained_draw" ,
643
+ unconstrained. data_type ( ) . clone ( ) ,
644
+ true ,
645
+ ) ) ;
638
646
arrays. push ( unconstrained. as_box ( ) ) ;
639
647
}
640
648
0 commit comments