Coverage report: 90%

Files Functions Classes

coverage.py v7.9.1, created at 2025-06-27 14:46 +0000

File function statements missing excluded branches partial coverage
src/bartz/BART.py DataFrame.to_numpy 1 1 0 0 0 0%
src/bartz/BART.py Series.to_numpy 1 1 0 0 0 0%
src/bartz/BART.py gbart.__init__ 28 0 0 8 0 100%
src/bartz/BART.py gbart.prob_test 3 0 0 2 0 100%
src/bartz/BART.py gbart.prob_test_mean 3 0 0 2 0 100%
src/bartz/BART.py gbart.prob_train 3 0 0 2 0 100%
src/bartz/BART.py gbart.prob_train_mean 3 0 0 2 0 100%
src/bartz/BART.py gbart.sigma_mean 3 0 0 2 0 100%
src/bartz/BART.py gbart.varcount 1 0 0 0 0 100%
src/bartz/BART.py gbart.varcount_mean 1 0 0 0 0 100%
src/bartz/BART.py gbart.yhat_test_mean 3 0 0 2 0 100%
src/bartz/BART.py gbart.yhat_train 2 0 0 0 0 100%
src/bartz/BART.py gbart.yhat_train_mean 3 0 0 2 0 100%
src/bartz/BART.py gbart.predict 6 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 29 9 0 16 4 67%
src/bartz/BART.py gbart._check_type_settings 15 7 0 10 4 56%
src/bartz/BART.py gbart._process_offset_settings 10 1 0 6 1 88%
src/bartz/BART.py gbart._process_leaf_sdev_settings 7 0 0 6 1 92%
src/bartz/BART.py gbart._determine_splits 8 0 0 6 0 100%
src/bartz/BART.py gbart._bin_predictors 1 0 0 0 0 100%
src/bartz/BART.py gbart._setup_mcmc 20 5 0 10 1 73%
src/bartz/BART.py gbart._run_mcmc 9 1 0 6 1 87%
src/bartz/BART.py gbart._extract_sigma 4 0 0 2 0 100%
src/bartz/BART.py gbart._predict 1 0 0 0 0 100%
src/bartz/BART.py (no function) 77 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 format_tree 10 0 0 0 0 100%
src/bartz/debug.py format_tree.traverse_tree 33 7 0 16 2 73%
src/bartz/debug.py tree_actual_depth 4 0 0 0 0 100%
src/bartz/debug.py forest_depth_distr 3 0 0 0 0 100%
src/bartz/debug.py trace_depth_distr 1 0 0 0 0 100%
src/bartz/debug.py points_per_decision_node_distr 6 0 0 0 0 100%
src/bartz/debug.py forest_points_per_decision_node_distr 4 0 0 0 0 100%
src/bartz/debug.py forest_points_per_decision_node_distr.loop 1 0 0 0 0 100%
src/bartz/debug.py trace_points_per_decision_node_distr 3 0 0 0 0 100%
src/bartz/debug.py trace_points_per_decision_node_distr.loop 1 0 0 0 0 100%
src/bartz/debug.py points_per_leaf_distr 5 0 0 0 0 100%
src/bartz/debug.py forest_points_per_leaf_distr 4 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 2 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_rule_consistency 8 1 0 2 1 80%
src/bartz/debug.py check_rule_consistency._check_recursive 9 0 0 2 0 100%
src/bartz/debug.py check_num_nodes 4 0 0 0 0 100%
src/bartz/debug.py check_var_in_bounds 3 0 0 0 0 100%
src/bartz/debug.py check_split_in_bounds 2 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 3 0 0 0 0 100%
src/bartz/debug.py _get_next_line 4 1 0 2 1 67%
src/bartz/debug.py scan_BART_trees 39 8 0 14 4 77%
src/bartz/debug.py TraceWithOffset.from_trees_trace 2 0 0 0 0 100%
src/bartz/debug.py trees_BART_to_bartz 26 0 0 8 0 100%
src/bartz/debug.py SamplePriorStack.initial 2 0 0 0 0 100%
src/bartz/debug.py SamplePriorTrees.initial 2 0 0 0 0 100%
src/bartz/debug.py SamplePriorCarry.initial 2 0 0 0 0 100%
src/bartz/debug.py SamplePriorX.initial 6 0 0 0 0 100%
src/bartz/debug.py SamplePriorX._sequence 6 0 0 2 0 100%
src/bartz/debug.py sample_prior_onetree 5 0 0 0 0 100%
src/bartz/debug.py sample_prior_onetree.loop 21 0 0 0 0 100%
src/bartz/debug.py sample_prior_onetree.loop.write_push_stack 1 0 0 0 0 100%
src/bartz/debug.py sample_prior_onetree.loop.pop_push_stack 5 0 0 0 0 100%
src/bartz/debug.py sample_prior_forest 1 0 0 0 0 100%
src/bartz/debug.py sample_prior 3 0 0 0 0 100%
src/bartz/debug.py debug_gbart.__init__ 5 0 0 2 0 100%
src/bartz/debug.py debug_gbart.show_tree 4 4 0 0 0 0%
src/bartz/debug.py debug_gbart.sigma_harmonic_mean 12 12 0 2 0 0%
src/bartz/debug.py debug_gbart.compare_resid 8 0 0 2 0 100%
src/bartz/debug.py debug_gbart.avg_acc 3 3 0 0 0 0%
src/bartz/debug.py debug_gbart.avg_acc.acc 3 3 0 0 0 0%
src/bartz/debug.py debug_gbart.avg_prop 6 6 0 0 0 0%
src/bartz/debug.py debug_gbart.avg_prop.prop 1 1 0 0 0 0%
src/bartz/debug.py debug_gbart.avg_move 3 3 0 0 0 0%
src/bartz/debug.py debug_gbart.depth_distr 1 0 0 0 0 100%
src/bartz/debug.py debug_gbart.points_per_decision_node_distr 1 0 0 0 0 100%
src/bartz/debug.py debug_gbart.points_per_leaf_distr 1 0 0 0 0 100%
src/bartz/debug.py debug_gbart.check_trees 1 0 0 0 0 100%
src/bartz/debug.py debug_gbart.tree_goes_bad 3 3 0 0 0 0%
src/bartz/debug.py (no function) 117 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 var_histogram 2 0 0 0 0 100%
src/bartz/grove.py (no function) 26 0 0 0 0 100%
src/bartz/jaxext/__init__.py vmap_nodoc 4 0 0 0 0 100%
src/bartz/jaxext/__init__.py minimal_unsigned_dtype 7 3 0 6 1 54%
src/bartz/jaxext/__init__.py unique 9 0 0 4 0 100%
src/bartz/jaxext/__init__.py unique.loop 4 0 0 0 0 100%
src/bartz/jaxext/__init__.py split.__init__ 1 0 0 0 0 100%
src/bartz/jaxext/__init__.py split.__len__ 1 0 0 0 0 100%
src/bartz/jaxext/__init__.py split.pop 11 0 0 6 1 94%
src/bartz/jaxext/__init__.py truncated_normal_onesided 12 0 0 0 0 100%
src/bartz/jaxext/__init__.py (no function) 20 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py expand_axes 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py expand_axes.expand_axis 1 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py check_no_nones 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py check_no_nones.check_not_none 1 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py extract_size 5 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py extract_size.get_size 3 0 0 2 0 100%
src/bartz/jaxext/_autobatch.py sum_nbytes 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py sum_nbytes.nbytes 1 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py next_divisor_small 4 1 0 4 2 62%
src/bartz/jaxext/_autobatch.py next_divisor_large 5 0 0 4 1 89%
src/bartz/jaxext/_autobatch.py next_divisor 5 0 0 4 0 100%
src/bartz/jaxext/_autobatch.py pull_nonbatched 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py pull_nonbatched.pull_nonbatched 3 0 0 2 0 100%
src/bartz/jaxext/_autobatch.py push_nonbatched 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py push_nonbatched.push_nonbatched 3 0 0 2 0 100%
src/bartz/jaxext/_autobatch.py move_axes_out 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py move_axes_out.move_axis_out 1 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py move_axes_in 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py move_axes_in.move_axis_in 1 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py batch 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py batch.batch 1 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py unbatch 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py unbatch.unbatch 1 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py check_same 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py check_same.check_same 2 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py autobatch 6 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py autobatch.batched_func 28 0 0 4 0 100%
src/bartz/jaxext/_autobatch.py autobatch.batched_func.loop 5 0 0 0 0 100%
src/bartz/jaxext/_autobatch.py (no function) 26 0 0 0 0 100%
src/bartz/jaxext/scipy/__init__.py (no function) 0 0 0 0 0 100%
src/bartz/jaxext/scipy/special.py _float_type 2 0 0 0 0 100%
src/bartz/jaxext/scipy/special.py _castto 2 0 0 0 0 100%
src/bartz/jaxext/scipy/special.py _castto.newfunc 1 0 0 0 0 100%
src/bartz/jaxext/scipy/special.py gammainccinv 7 0 0 0 0 100%
src/bartz/jaxext/scipy/special.py ndtri 5 2 0 2 1 57%
src/bartz/jaxext/scipy/special.py _ndtri 27 0 0 0 0 100%
src/bartz/jaxext/scipy/special.py _ndtri._create_polynomial 4 0 0 2 0 100%
src/bartz/jaxext/scipy/special.py (no function) 12 0 0 0 0 100%
src/bartz/jaxext/scipy/stats.py invgamma.ppf 1 0 0 0 0 100%
src/bartz/jaxext/scipy/stats.py (no function) 4 0 0 0 0 100%
src/bartz/mcmcloop.py BurninTrace.from_state 1 0 0 0 0 100%
src/bartz/mcmcloop.py MainTrace.from_state 1 0 0 0 0 100%
src/bartz/mcmcloop.py Callback.__call__ 1 1 0 0 0 0%
src/bartz/mcmcloop.py run_mcmc 13 0 0 6 0 100%
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 5 0 0 0 0 100%
src/bartz/mcmcloop.py _run_mcmc_inner_loop.loop_impl 14 2 0 4 1 83%
src/bartz/mcmcloop.py _run_mcmc_inner_loop.loop_impl.save_to_burnin_trace 1 0 0 0 0 100%
src/bartz/mcmcloop.py _run_mcmc_inner_loop.loop_impl.save_to_main_trace 2 0 0 0 0 100%
src/bartz/mcmcloop.py _run_mcmc_inner_loop.loop_noop 1 0 0 0 0 100%
src/bartz/mcmcloop.py _run_mcmc_inner_loop.loop 2 0 0 0 0 100%
src/bartz/mcmcloop.py pytree_at_set 2 0 0 0 0 100%
src/bartz/mcmcloop.py pytree_at_set.at_set 3 0 0 2 0 100%
src/bartz/mcmcloop.py make_print_callback 2 0 0 0 0 100%
src/bartz/mcmcloop.py make_print_callback.asarray_or_none 1 0 0 0 0 100%
src/bartz/mcmcloop.py _print_callback 7 0 0 4 2 82%
src/bartz/mcmcloop.py _print_callback.print_report 1 0 0 0 0 100%
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 2 0 0 0 0%
src/bartz/mcmcloop.py _convert_jax_arrays_in_args.convert_jax_arrays.convert_jax_arrays 5 5 0 4 0 0%
src/bartz/mcmcloop.py _convert_jax_arrays_in_args.new_func 3 3 0 0 0 0%
src/bartz/mcmcloop.py _print_report 8 8 0 0 0 0%
src/bartz/mcmcloop.py _print_report.acc_string 3 3 0 2 0 0%
src/bartz/mcmcloop.py TreesTrace.from_dataclass 1 0 0 0 0 100%
src/bartz/mcmcloop.py evaluate_trace 6 0 0 0 0 100%
src/bartz/mcmcloop.py evaluate_trace.loop 3 0 0 0 0 100%
src/bartz/mcmcloop.py compute_varcount 2 0 0 0 0 100%
src/bartz/mcmcloop.py (no function) 64 0 0 0 0 100%
src/bartz/mcmcstep.py init 23 2 0 6 1 90%
src/bartz/mcmcstep.py init.make_forest 1 0 0 0 0 100%
src/bartz/mcmcstep.py _choose_suffstat_batch_size 22 10 0 12 2 53%
src/bartz/mcmcstep.py _choose_suffstat_batch_size.get_platform 9 2 0 2 1 73%
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 11 0 0 0 0 100%
src/bartz/mcmcstep.py propose_grow_moves 13 0 0 0 0 100%
src/bartz/mcmcstep.py choose_leaf 9 0 0 0 0 100%
src/bartz/mcmcstep.py growable_leaves 1 0 0 0 0 100%
src/bartz/mcmcstep.py categorical 3 0 0 0 0 100%
src/bartz/mcmcstep.py choose_variable 4 0 0 2 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 2 0 0 0 0 100%
src/bartz/mcmcstep.py compute_partial_ratio 6 0 0 0 0 100%
src/bartz/mcmcstep.py propose_prune_moves 3 0 0 0 0 100%
src/bartz/mcmcstep.py choose_leaf_parent 11 0 0 0 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 21 0 0 8 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 13 0 0 0 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 6 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 22 0 0 4 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 5 0 0 0 0 100%
src/bartz/mcmcstep.py (no function) 167 0 0 0 0 100%
src/bartz/prepcovars.py parse_xinfo 6 0 0 0 0 100%
src/bartz/prepcovars.py quantilized_splits_from_matrix 7 0 0 2 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 4 0 100%
src/bartz/prepcovars.py _huge_value 3 0 0 2 0 100%
src/bartz/prepcovars.py _ensure_unsigned 1 0 0 0 0 100%
src/bartz/prepcovars.py _signed_to_unsigned 15 8 0 10 4 44%
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) 17 0 0 0 0 100%
Total   1757 135 0 278 38 90%

No items found using the specified filter.