Coverage report:
85%
hide covered
Files
Functions
Classes
coverage.py v7.8.2
, created at 2025-05-29 23:01 +0000
File
function
statements
missing
excluded
branches
partial
coverage
src/bartz/BART.py
gbart.__init__
28
0
0
4
0
100%
src/bartz/BART.py
gbart.yhat_train
2
0
0
0
0
100%
src/bartz/BART.py
gbart.yhat_train_mean
1
0
0
0
0
100%
src/bartz/BART.py
gbart.predict
5
0
0
2
0
100%
src/bartz/BART.py
gbart._process_predictor_input
7
0
0
2
0
100%
src/bartz/BART.py
gbart._process_response_input
7
0
0
2
0
100%
src/bartz/BART.py
gbart._check_same_length
2
0
0
0
0
100%
src/bartz/BART.py
gbart._process_error_variance_settings
26
6
0
16
4
71%
src/bartz/BART.py
gbart._process_type_settings
12
3
0
10
2
77%
src/bartz/BART.py
gbart._process_offset_settings
8
1
0
6
1
86%
src/bartz/BART.py
gbart._process_leaf_sdev_settings
7
0
0
6
1
92%
src/bartz/BART.py
gbart._determine_splits
3
0
0
2
0
100%
src/bartz/BART.py
gbart._bin_predictors
1
0
0
0
0
100%
src/bartz/BART.py
gbart._setup_mcmc
11
0
0
4
1
93%
src/bartz/BART.py
gbart._run_mcmc
9
2
0
6
2
73%
src/bartz/BART.py
gbart._extract_sigma
3
0
0
2
0
100%
src/bartz/BART.py
gbart._predict
1
0
0
0
0
100%
src/bartz/BART.py
gbart._show_tree
6
6
0
0
0
0%
src/bartz/BART.py
gbart._sigma_harmonic_mean
10
10
0
2
0
0%
src/bartz/BART.py
gbart._compare_resid
8
0
0
2
0
100%
src/bartz/BART.py
gbart._avg_acc
3
3
0
0
0
0%
src/bartz/BART.py
gbart._avg_acc.acc
3
3
0
0
0
0%
src/bartz/BART.py
gbart._avg_prop
6
6
0
0
0
0%
src/bartz/BART.py
gbart._avg_prop.prop
1
1
0
0
0
0%
src/bartz/BART.py
gbart._avg_move
3
3
0
0
0
0%
src/bartz/BART.py
gbart._depth_distr
4
4
0
0
0
0%
src/bartz/BART.py
gbart._points_per_leaf_distr
2
0
0
0
0
100%
src/bartz/BART.py
gbart._check_trees
2
0
0
0
0
100%
src/bartz/BART.py
gbart._tree_goes_bad
3
3
0
0
0
0%
src/bartz/BART.py
(no function)
52
0
0
0
0
100%
src/bartz/__init__.py
(no function)
2
0
0
0
0
100%
src/bartz/_version.py
(no function)
1
0
0
0
0
100%
src/bartz/debug.py
print_tree
8
8
0
0
0
0%
src/bartz/debug.py
print_tree.traverse_tree
33
33
0
16
0
0%
src/bartz/debug.py
tree_actual_depth
4
4
0
0
0
0%
src/bartz/debug.py
forest_depth_distr
3
3
0
0
0
0%
src/bartz/debug.py
trace_depth_distr
1
1
0
0
0
0%
src/bartz/debug.py
points_per_leaf_distr
6
0
0
0
0
100%
src/bartz/debug.py
forest_points_per_leaf_distr
5
0
0
0
0
100%
src/bartz/debug.py
forest_points_per_leaf_distr.loop
1
0
0
0
0
100%
src/bartz/debug.py
trace_points_per_leaf_distr
3
0
0
0
0
100%
src/bartz/debug.py
trace_points_per_leaf_distr.loop
1
0
0
0
0
100%
src/bartz/debug.py
check_types
3
0
0
0
0
100%
src/bartz/debug.py
check_sizes
1
0
0
0
0
100%
src/bartz/debug.py
check_unused_node
1
0
0
0
0
100%
src/bartz/debug.py
check_leaf_values
1
0
0
0
0
100%
src/bartz/debug.py
check_stray_nodes
7
0
0
0
0
100%
src/bartz/debug.py
check_tree
8
0
0
2
0
100%
src/bartz/debug.py
describe_error
1
1
0
0
0
0%
src/bartz/debug.py
check_trace
1
0
0
0
0
100%
src/bartz/debug.py
(no function)
23
0
0
0
0
100%
src/bartz/grove.py
make_tree
1
0
0
0
0
100%
src/bartz/grove.py
tree_depth
1
0
0
0
0
100%
src/bartz/grove.py
traverse_tree
5
0
0
0
0
100%
src/bartz/grove.py
traverse_tree.loop
7
0
0
0
0
100%
src/bartz/grove.py
traverse_forest
1
0
0
0
0
100%
src/bartz/grove.py
evaluate_forest
7
0
0
2
0
100%
src/bartz/grove.py
is_actual_leaf
10
0
0
2
0
100%
src/bartz/grove.py
is_leaves_parent
7
0
0
0
0
100%
src/bartz/grove.py
tree_depths
8
0
0
4
0
100%
src/bartz/grove.py
is_used
4
0
0
0
0
100%
src/bartz/grove.py
forest_fill
4
0
0
0
0
100%
src/bartz/grove.py
(no function)
18
0
0
0
0
100%
src/bartz/jaxext.py
float_type
2
0
0
0
0
100%
src/bartz/jaxext.py
_castto
2
0
0
0
0
100%
src/bartz/jaxext.py
_castto.newfunc
1
0
0
0
0
100%
src/bartz/jaxext.py
scipy.special.gammainccinv
7
0
0
0
0
100%
src/bartz/jaxext.py
scipy.stats.invgamma.ppf
1
0
0
0
0
100%
src/bartz/jaxext.py
vmap_nodoc
4
0
0
0
0
100%
src/bartz/jaxext.py
huge_value
3
0
0
2
0
100%
src/bartz/jaxext.py
minimal_unsigned_dtype
7
3
0
6
1
54%
src/bartz/jaxext.py
signed_to_unsigned
11
5
0
10
4
48%
src/bartz/jaxext.py
ensure_unsigned
1
0
0
0
0
100%
src/bartz/jaxext.py
unique
9
0
0
4
0
100%
src/bartz/jaxext.py
unique.loop
4
0
0
0
0
100%
src/bartz/jaxext.py
autobatch
20
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.expand_axes
3
0
0
2
0
100%
src/bartz/jaxext.py
autobatch.check_no_nones
2
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.check_no_nones.check_not_none
1
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.extract_size
5
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.extract_size.get_size
3
0
0
2
0
100%
src/bartz/jaxext.py
autobatch.sum_nbytes
2
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.sum_nbytes.nbytes
1
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.next_divisor_small
4
1
0
4
2
62%
src/bartz/jaxext.py
autobatch.next_divisor_large
5
0
0
4
1
89%
src/bartz/jaxext.py
autobatch.next_divisor
5
0
0
4
0
100%
src/bartz/jaxext.py
autobatch.pull_nonbatched
2
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.pull_nonbatched.pull_nonbatched
3
0
0
2
0
100%
src/bartz/jaxext.py
autobatch.push_nonbatched
2
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.push_nonbatched.push_nonbatched
3
0
0
2
0
100%
src/bartz/jaxext.py
autobatch.move_axes_out
2
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.move_axes_out.move_axis_out
1
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.move_axes_in
2
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.move_axes_in.move_axis_in
1
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.batch
2
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.batch.batch
1
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.unbatch
2
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.unbatch.unbatch
1
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.check_same
2
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.check_same.check_same
2
0
0
0
0
100%
src/bartz/jaxext.py
autobatch.batched_func
27
0
0
4
0
100%
src/bartz/jaxext.py
autobatch.batched_func.loop
5
0
0
0
0
100%
src/bartz/jaxext.py
split.__init__
1
0
0
0
0
100%
src/bartz/jaxext.py
split.__len__
1
0
0
0
0
100%
src/bartz/jaxext.py
split.pop
10
0
0
6
1
94%
src/bartz/jaxext.py
(no function)
30
0
0
0
0
100%
src/bartz/mcmcloop.py
default_onlymain_extractor
1
0
0
0
0
100%
src/bartz/mcmcloop.py
default_both_extractor
1
0
0
0
0
100%
src/bartz/mcmcloop.py
run_mcmc
28
4
0
12
2
85%
src/bartz/mcmcloop.py
run_mcmc.empty_trace
1
0
0
0
0
100%
src/bartz/mcmcloop.py
_compute_i_skip
2
0
0
0
0
100%
src/bartz/mcmcloop.py
_run_mcmc_inner_loop
3
0
0
0
0
100%
src/bartz/mcmcloop.py
_run_mcmc_inner_loop.loop
18
1
0
4
1
91%
src/bartz/mcmcloop.py
_run_mcmc_inner_loop.loop.update_trace
2
0
0
0
0
100%
src/bartz/mcmcloop.py
_run_mcmc_inner_loop.loop.update_trace.assign_at_index
3
0
0
2
0
100%
src/bartz/mcmcloop.py
make_print_callbacks
1
0
0
0
0
100%
src/bartz/mcmcloop.py
_print_callback_inner
4
0
0
2
1
83%
src/bartz/mcmcloop.py
_print_dot
2
2
0
2
0
0%
src/bartz/mcmcloop.py
_print_callback_outer
7
1
0
4
2
73%
src/bartz/mcmcloop.py
_convert_jax_arrays_in_args
4
0
0
0
0
100%
src/bartz/mcmcloop.py
_convert_jax_arrays_in_args.convert_jax_arrays
2
0
0
0
0
100%
src/bartz/mcmcloop.py
_convert_jax_arrays_in_args.convert_jax_arrays.convert_jax_arrays
5
2
0
4
2
56%
src/bartz/mcmcloop.py
_convert_jax_arrays_in_args.new_func
3
0
0
0
0
100%
src/bartz/mcmcloop.py
_print_report
13
0
0
6
1
95%
src/bartz/mcmcloop.py
_print_report.acc_string
3
0
0
2
0
100%
src/bartz/mcmcloop.py
evaluate_trace
5
0
0
0
0
100%
src/bartz/mcmcloop.py
evaluate_trace.loop
2
0
0
0
0
100%
src/bartz/mcmcloop.py
(no function)
23
0
0
0
0
100%
src/bartz/mcmcstep.py
init
18
1
0
4
1
91%
src/bartz/mcmcstep.py
init.make_forest
1
0
0
0
0
100%
src/bartz/mcmcstep.py
_choose_suffstat_batch_size
22
18
0
12
2
18%
src/bartz/mcmcstep.py
_choose_suffstat_batch_size.get_platform
8
8
0
2
0
0%
src/bartz/mcmcstep.py
step
8
0
0
2
0
100%
src/bartz/mcmcstep.py
step_trees
3
0
0
0
0
100%
src/bartz/mcmcstep.py
propose_moves
12
0
0
0
0
100%
src/bartz/mcmcstep.py
propose_grow_moves
7
0
0
0
0
100%
src/bartz/mcmcstep.py
choose_leaf
9
0
0
0
0
100%
src/bartz/mcmcstep.py
growable_leaves
4
0
0
2
0
100%
src/bartz/mcmcstep.py
categorical
3
0
0
0
0
100%
src/bartz/mcmcstep.py
choose_variable
2
0
0
0
0
100%
src/bartz/mcmcstep.py
fully_used_variables
5
0
0
0
0
100%
src/bartz/mcmcstep.py
ancestor_variables
6
0
0
0
0
100%
src/bartz/mcmcstep.py
ancestor_variables.loop
6
0
0
0
0
100%
src/bartz/mcmcstep.py
split_range
6
0
0
0
0
100%
src/bartz/mcmcstep.py
split_range.loop
8
0
0
0
0
100%
src/bartz/mcmcstep.py
randint_exclude
6
0
0
0
0
100%
src/bartz/mcmcstep.py
randint_exclude.loop
1
0
0
0
0
100%
src/bartz/mcmcstep.py
choose_split
3
0
0
0
0
100%
src/bartz/mcmcstep.py
compute_partial_ratio
8
0
0
0
0
100%
src/bartz/mcmcstep.py
propose_prune_moves
4
0
0
0
0
100%
src/bartz/mcmcstep.py
choose_leaf_parent
11
0
0
2
0
100%
src/bartz/mcmcstep.py
randint_masked
3
0
0
0
0
100%
src/bartz/mcmcstep.py
accept_moves_and_sample_leaves
3
0
0
0
0
100%
src/bartz/mcmcstep.py
accept_moves_parallel_stage
16
0
0
6
0
100%
src/bartz/mcmcstep.py
apply_grow_to_indices
5
0
0
0
0
100%
src/bartz/mcmcstep.py
compute_count_trees
9
0
0
0
0
100%
src/bartz/mcmcstep.py
count_datapoints_per_leaf
3
0
0
2
0
100%
src/bartz/mcmcstep.py
_count_scan
3
0
0
0
0
100%
src/bartz/mcmcstep.py
_count_scan.loop
1
0
0
0
0
100%
src/bartz/mcmcstep.py
_aggregate_scatter
1
0
0
0
0
100%
src/bartz/mcmcstep.py
_count_vec
1
0
0
0
0
100%
src/bartz/mcmcstep.py
_aggregate_batched_alltrees
5
0
0
0
0
100%
src/bartz/mcmcstep.py
compute_prec_trees
9
0
0
0
0
100%
src/bartz/mcmcstep.py
prec_per_leaf
3
1
0
2
1
60%
src/bartz/mcmcstep.py
_prec_scan
3
3
0
0
0
0%
src/bartz/mcmcstep.py
_prec_scan.loop
1
1
0
0
0
0%
src/bartz/mcmcstep.py
_prec_vec
1
0
0
0
0
100%
src/bartz/mcmcstep.py
complete_ratio
2
0
0
0
0
100%
src/bartz/mcmcstep.py
compute_p_prune
11
0
0
2
0
100%
src/bartz/mcmcstep.py
adapt_leaf_trees_to_grow_indices
2
0
0
0
0
100%
src/bartz/mcmcstep.py
precompute_likelihood_terms
5
0
0
0
0
100%
src/bartz/mcmcstep.py
precompute_leaf_terms
5
0
0
0
0
100%
src/bartz/mcmcstep.py
accept_moves_sequential_stage
7
0
0
0
0
100%
src/bartz/mcmcstep.py
accept_moves_sequential_stage.loop
2
0
0
0
0
100%
src/bartz/mcmcstep.py
accept_move_and_sample_leaves
31
0
0
6
0
100%
src/bartz/mcmcstep.py
sum_resid
4
0
0
2
0
100%
src/bartz/mcmcstep.py
_aggregate_batched_onetree
4
0
0
0
0
100%
src/bartz/mcmcstep.py
compute_likelihood_ratio
2
0
0
0
0
100%
src/bartz/mcmcstep.py
accept_moves_final_stage
1
0
0
0
0
100%
src/bartz/mcmcstep.py
apply_moves_to_leaf_indices
3
0
0
0
0
100%
src/bartz/mcmcstep.py
apply_moves_to_split_trees
2
0
0
0
0
100%
src/bartz/mcmcstep.py
step_sigma
9
0
0
2
0
100%
src/bartz/mcmcstep.py
step_z
6
0
0
0
0
100%
src/bartz/mcmcstep.py
(no function)
165
0
0
0
0
100%
src/bartz/prepcovars.py
quantilized_splits_from_matrix
4
0
0
0
0
100%
src/bartz/prepcovars.py
quantilized_splits_from_matrix.quantilize
1
0
0
0
0
100%
src/bartz/prepcovars.py
_quantilized_splits_from_matrix
16
0
0
2
0
100%
src/bartz/prepcovars.py
uniform_splits_from_matrix
6
0
0
0
0
100%
src/bartz/prepcovars.py
bin_predictors
4
0
0
0
0
100%
src/bartz/prepcovars.py
bin_predictors.bin_predictors
2
0
0
0
0
100%
src/bartz/prepcovars.py
(no function)
12
0
0
0
0
100%
Total
1293
152
0
230
33
85%
No items found using the specified filter.