Coverage report: 85%

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.