Coverage report: 93%

Files Functions Classes

coverage.py v7.13.0, created at 2026-01-13 00:35 +0000

      Statements   Branches   Total
File function   coverage statements missing excluded   coverage branches partial   coverage
src / bartz / BART / __init__.py (no function)   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.__init__   100% 3 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.ndpost   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.offset   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.sigest   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.yhat_test   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart._main_trace   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart._burnin_trace   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart._mcmc_state   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart._splits   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart._x_train_fmt   0% 1 1 0   100% 0 0   0%
src / bartz / BART / _gbart.py mc_gbart.prob_test   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.prob_test_mean   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.prob_train   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.prob_train_mean   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.sigma   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.sigma_   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.sigma_mean   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.varcount   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.varcount_mean   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.varprob   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.varprob_mean   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.yhat_test_mean   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.yhat_train   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.yhat_train_mean   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py mc_gbart.predict   100% 1 0 0   100% 0 0   100%
src / bartz / BART / _gbart.py gbart.__init__   60% 5 2 0   50% 2 1   57%
src / bartz / BART / _gbart.py process_mc_cores   74% 23 6 0   50% 18 5   63%
src / bartz / BART / _gbart.py get_platform   67% 6 2 0   75% 4 1   70%
src / bartz / BART / _gbart.py (no function)   100% 66 0 0   100% 0 0   100%
src / bartz / __init__.py (no function)   100% 4 0 0   100% 0 0   100%
src / bartz / _interface.py DataFrame.to_numpy   100% 0 0 1   100% 0 0   100%
src / bartz / _interface.py Series.to_numpy   100% 0 0 1   100% 0 0   100%
src / bartz / _interface.py Bart.__init__   100% 28 0 0   100% 8 0   100%
src / bartz / _interface.py Bart.ndpost   100% 1 0 0   100% 0 0   100%
src / bartz / _interface.py Bart.prob_test   100% 3 0 0   100% 2 0   100%
src / bartz / _interface.py Bart.prob_test_mean   100% 3 0 0   100% 2 0   100%
src / bartz / _interface.py Bart.prob_train   100% 3 0 0   100% 2 0   100%
src / bartz / _interface.py Bart.prob_train_mean   100% 3 0 0   100% 2 0   100%
src / bartz / _interface.py Bart.sigma   100% 4 0 0   100% 2 0   100%
src / bartz / _interface.py Bart.sigma_   100% 4 0 0   100% 2 0   100%
src / bartz / _interface.py Bart.sigma_mean   100% 3 0 0   100% 2 0   100%
src / bartz / _interface.py Bart.varcount   100% 3 0 0   100% 0 0   100%
src / bartz / _interface.py Bart.varcount_mean   100% 1 0 0   100% 0 0   100%
src / bartz / _interface.py Bart.varprob   100% 9 0 0   100% 2 0   100%
src / bartz / _interface.py Bart.varprob_mean   100% 1 0 0   100% 0 0   100%
src / bartz / _interface.py Bart.yhat_test_mean   100% 3 0 0   100% 2 0   100%
src / bartz / _interface.py Bart.yhat_train   100% 2 0 0   100% 0 0   100%
src / bartz / _interface.py Bart.yhat_train_mean   100% 3 0 0   100% 2 0   100%
src / bartz / _interface.py Bart.predict   100% 6 0 0   100% 2 0   100%
src / bartz / _interface.py Bart._process_predictor_input   100% 7 0 0   100% 2 0   100%
src / bartz / _interface.py Bart._process_response_input   100% 5 0 0   100% 2 0   100%
src / bartz / _interface.py Bart._check_same_length   100% 2 0 0   100% 0 0   100%
src / bartz / _interface.py Bart._process_error_variance_settings   62% 24 9 0   62% 16 4   62%
src / bartz / _interface.py Bart._linear_regression   100% 6 0 0   100% 0 0   100%
src / bartz / _interface.py Bart._check_type_settings   53% 15 7 0   60% 10 4   56%
src / bartz / _interface.py Bart._process_sparsity_settings   100% 8 0 0   83% 6 1   93%
src / bartz / _interface.py Bart._process_offset_settings   90% 10 1 0   83% 6 1   88%
src / bartz / _interface.py Bart._process_leaf_sdev_settings   100% 7 0 0   83% 6 1   92%
src / bartz / _interface.py Bart._determine_splits   100% 8 0 0   100% 6 0   100%
src / bartz / _interface.py Bart._bin_predictors   100% 1 0 0   100% 0 0   100%
src / bartz / _interface.py Bart._setup_mcmc   80% 25 5 0   75% 12 1   78%
src / bartz / _interface.py Bart._run_mcmc   92% 12 1 0   83% 6 1   89%
src / bartz / _interface.py Bart._predict   100% 2 0 0   100% 0 0   100%
src / bartz / _interface.py process_device_settings   86% 22 3 0   83% 12 2   85%
src / bartz / _interface.py (no function)   100% 90 0 4   100% 0 0   100%
src / bartz / _profiler.py get_profile_mode   100% 1 0 0   100% 0 0   100%
src / bartz / _profiler.py set_profile_mode   100% 1 0 0   100% 0 0   100%
src / bartz / _profiler.py profile_mode   100% 5 0 1   100% 0 0   100%
src / bartz / _profiler.py jit_and_block_if_profiling   100% 6 0 0   100% 0 0   100%
src / bartz / _profiler.py jit_and_block_if_profiling.jab_inner_wrapper   100% 3 0 0   100% 0 0   100%
src / bartz / _profiler.py jit_and_block_if_profiling.jab_outer_wrapper   80% 5 1 0   75% 4 1   78%
src / bartz / _profiler.py jit_if_profiling   100% 4 0 0   100% 0 0   100%
src / bartz / _profiler.py jit_if_profiling.wrapper   100% 3 0 0   100% 2 0   100%
src / bartz / _profiler.py jit_if_not_profiling   100% 4 0 0   100% 0 0   100%
src / bartz / _profiler.py jit_if_not_profiling.wrapper   100% 3 0 0   100% 2 0   100%
src / bartz / _profiler.py scan_if_not_profiling   100% 7 0 0   100% 4 0   100%
src / bartz / _profiler.py cond_if_not_profiling   100% 5 0 0   100% 4 0   100%
src / bartz / _profiler.py callback_if_not_profiling   0% 3 3 0   0% 2 0   0%
src / bartz / _profiler.py vmap_chains_if_profiling   100% 4 0 0   100% 0 0   100%
src / bartz / _profiler.py vmap_chains_if_profiling.wrapper   100% 3 0 0   100% 2 0   100%
src / bartz / _profiler.py vmap_chains_if_not_profiling   100% 4 0 0   100% 0 0   100%
src / bartz / _profiler.py vmap_chains_if_not_profiling.wrapper   100% 3 0 0   100% 2 0   100%
src / bartz / _profiler.py (no function)   100% 24 0 0   100% 0 0   100%
src / bartz / _version.py (no function)   100% 1 0 0   100% 0 0   100%
src / bartz / debug.py format_tree   100% 10 0 0   100% 0 0   100%
src / bartz / debug.py format_tree.traverse_tree   79% 33 7 0   62% 16 2   73%
src / bartz / debug.py tree_actual_depth   100% 4 0 0   100% 0 0   100%
src / bartz / debug.py forest_depth_distr   100% 3 0 0   100% 0 0   100%
src / bartz / debug.py points_per_node_distr   100% 11 0 0   100% 4 0   100%
src / bartz / debug.py points_per_node_distr.func   100% 10 0 0   100% 4 0   100%
src / bartz / debug.py points_per_node_distr.func.count_points   88% 8 1 0   75% 4 1   83%
src / bartz / debug.py points_per_node_distr.func.count_nodes   100% 1 0 0   100% 0 0   100%
src / bartz / debug.py check   100% 2 0 0   100% 0 0   100%
src / bartz / debug.py check_types   100% 3 0 0   100% 0 0   100%
src / bartz / debug.py check_sizes   100% 1 0 0   100% 0 0   100%
src / bartz / debug.py check_unused_node   100% 1 0 0   100% 0 0   100%
src / bartz / debug.py check_leaf_values   100% 1 0 0   100% 0 0   100%
src / bartz / debug.py check_stray_nodes   100% 7 0 0   100% 0 0   100%
src / bartz / debug.py check_rule_consistency   89% 9 1 0   50% 2 1   82%
src / bartz / debug.py check_rule_consistency._check_recursive   100% 10 0 0   100% 2 0   100%
src / bartz / debug.py check_num_nodes   100% 4 0 0   100% 0 0   100%
src / bartz / debug.py check_var_in_bounds   100% 3 0 0   100% 0 0   100%
src / bartz / debug.py check_split_in_bounds   100% 2 0 0   100% 0 0   100%
src / bartz / debug.py check_tree   100% 8 0 0   100% 2 0   100%
src / bartz / debug.py describe_error   100% 1 0 0   100% 0 0   100%
src / bartz / debug.py check_trace   100% 10 0 0   100% 2 0   100%
src / bartz / debug.py _get_next_line   75% 4 1 0   50% 2 1   67%
src / bartz / debug.py scan_BART_trees   79% 39 8 0   71% 14 4   77%
src / bartz / debug.py TraceWithOffset.from_trees_trace   100% 2 0 0   100% 0 0   100%
src / bartz / debug.py trees_BART_to_bartz   100% 26 0 0   100% 8 0   100%
src / bartz / debug.py SamplePriorStack.initial   100% 2 0 0   100% 0 0   100%
src / bartz / debug.py SamplePriorTrees.initial   100% 2 0 0   100% 0 0   100%
src / bartz / debug.py SamplePriorCarry.initial   100% 2 0 0   100% 0 0   100%
src / bartz / debug.py SamplePriorX.initial   100% 6 0 0   100% 0 0   100%
src / bartz / debug.py SamplePriorX._sequence   100% 6 0 0   100% 2 0   100%
src / bartz / debug.py sample_prior_onetree   100% 5 0 0   100% 0 0   100%
src / bartz / debug.py sample_prior_onetree.loop   100% 21 0 0   100% 0 0   100%
src / bartz / debug.py sample_prior_onetree.loop.write_push_stack   100% 1 0 0   100% 0 0   100%
src / bartz / debug.py sample_prior_onetree.loop.pop_push_stack   100% 5 0 0   100% 0 0   100%
src / bartz / debug.py sample_prior_forest   100% 1 0 0   100% 0 0   100%
src / bartz / debug.py sample_prior   100% 3 0 0   100% 0 0   100%
src / bartz / debug.py debug_mc_gbart.__init__   100% 5 0 0   100% 2 0   100%
src / bartz / debug.py debug_mc_gbart.print_tree   0% 4 4 0   100% 0 0   0%
src / bartz / debug.py debug_mc_gbart.sigma_harmonic_mean   0% 11 11 0   0% 2 0   0%
src / bartz / debug.py debug_mc_gbart.compare_resid   100% 9 0 0   100% 2 0   100%
src / bartz / debug.py debug_mc_gbart.avg_acc   0% 3 3 0   100% 0 0   0%
src / bartz / debug.py debug_mc_gbart.avg_acc.acc   0% 3 3 0   100% 0 0   0%
src / bartz / debug.py debug_mc_gbart.avg_prop   0% 6 6 0   100% 0 0   0%
src / bartz / debug.py debug_mc_gbart.avg_prop.prop   0% 1 1 0   100% 0 0   0%
src / bartz / debug.py debug_mc_gbart.avg_move   0% 3 3 0   100% 0 0   0%
src / bartz / debug.py debug_mc_gbart.depth_distr   100% 4 0 0   50% 2 1   83%
src / bartz / debug.py debug_mc_gbart._points_per_node_distr   100% 4 0 0   100% 2 0   100%
src / bartz / debug.py debug_mc_gbart.points_per_decision_node_distr   100% 1 0 0   100% 0 0   100%
src / bartz / debug.py debug_mc_gbart.points_per_leaf_distr   100% 1 0 0   100% 0 0   100%
src / bartz / debug.py debug_mc_gbart.check_trees   100% 4 0 0   100% 2 0   100%
src / bartz / debug.py debug_mc_gbart.tree_goes_bad   0% 3 3 0   100% 0 0   0%
src / bartz / debug.py (no function)   100% 95 0 0   100% 0 0   100%
src / bartz / grove.py make_tree   100% 2 0 0   100% 0 0   100%
src / bartz / grove.py tree_depth   100% 1 0 0   100% 0 0   100%
src / bartz / grove.py traverse_tree   100% 5 0 0   100% 0 0   100%
src / bartz / grove.py traverse_tree.loop   100% 7 0 0   100% 0 0   100%
src / bartz / grove.py traverse_forest   100% 1 0 0   100% 0 0   100%
src / bartz / grove.py evaluate_forest   100% 8 0 0   100% 0 0   100%
src / bartz / grove.py is_actual_leaf   100% 10 0 0   100% 2 0   100%
src / bartz / grove.py is_leaves_parent   100% 7 0 0   100% 0 0   100%
src / bartz / grove.py tree_depths   100% 8 0 0   100% 4 0   100%
src / bartz / grove.py is_used   100% 4 0 0   100% 0 0   100%
src / bartz / grove.py forest_fill   100% 4 0 0   100% 0 0   100%
src / bartz / grove.py var_histogram   100% 9 0 0   100% 4 0   100%
src / bartz / grove.py var_histogram.scatter_add   100% 1 0 0   100% 0 0   100%
src / bartz / grove.py (no function)   100% 30 0 0   100% 0 0   100%
src / bartz / jaxext / __init__.py vmap_nodoc   100% 4 0 0   100% 0 0   100%
src / bartz / jaxext / __init__.py minimal_unsigned_dtype   57% 7 3 0   50% 6 1   54%
src / bartz / jaxext / __init__.py unique   100% 9 0 0   100% 4 0   100%
src / bartz / jaxext / __init__.py unique.loop   100% 4 0 0   100% 0 0   100%
src / bartz / jaxext / __init__.py split.__init__   100% 6 0 0   100% 2 0   100%
src / bartz / jaxext / __init__.py split.__len__   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / __init__.py split.pop   100% 10 0 0   100% 6 0   100%
src / bartz / jaxext / __init__.py _split_unpack   100% 5 0 0   75% 4 1   89%
src / bartz / jaxext / __init__.py _split_shaped   67% 6 2 0   25% 4 1   50%
src / bartz / jaxext / __init__.py truncated_normal_onesided   100% 16 0 0   100% 2 0   100%
src / bartz / jaxext / __init__.py get_default_device   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / __init__.py get_device_count   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / __init__.py is_key   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / __init__.py (no function)   100% 29 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py expand_axes   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py expand_axes.expand_axis   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py normalize_axes   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py normalize_axes.normalize_axis   100% 3 0 0   100% 2 0   100%
src / bartz / jaxext / _autobatch.py check_no_nones   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py check_no_nones.check_not_none   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py remove_axis   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py remove_axis.remove_axis   100% 3 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py extract_size   100% 5 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py extract_size.get_size   100% 3 0 0   100% 2 0   100%
src / bartz / jaxext / _autobatch.py sum_nbytes   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py sum_nbytes.nbytes   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py next_divisor_small   100% 4 0 0   100% 4 0   100%
src / bartz / jaxext / _autobatch.py next_divisor_large   100% 5 0 0   100% 4 0   100%
src / bartz / jaxext / _autobatch.py next_divisor   100% 5 0 0   100% 4 0   100%
src / bartz / jaxext / _autobatch.py pull_nonbatched   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py pull_nonbatched.pull_nonbatched   100% 3 0 0   100% 2 0   100%
src / bartz / jaxext / _autobatch.py push_nonbatched   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py push_nonbatched.push_nonbatched   100% 3 0 0   100% 2 0   100%
src / bartz / jaxext / _autobatch.py move_axes_out   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py move_axes_out.move_axis_out   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py move_axes_in   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py move_axes_in.move_axis_in   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py batch   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py batch.batch   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py unbatch   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py unbatch.unbatch   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py reduce   100% 7 0 0   100% 2 0   100%
src / bartz / jaxext / _autobatch.py reduce.reduce   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py identity   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py identity.identity   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py reduction_dtype   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py identity_for   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py check_same   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py check_same.check_same   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py autobatch   100% 4 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py autobatch.autobatch_wrapper   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / _autobatch.py batched_func   100% 46 0 0   100% 16 0   100%
src / bartz / jaxext / _autobatch.py batching_loop   100% 8 0 0   100% 2 0   100%
src / bartz / jaxext / _autobatch.py (no function)   100% 41 0 0   100% 0 0   100%
src / bartz / jaxext / scipy / __init__.py (no function)   100% 0 0 0   100% 0 0   100%
src / bartz / jaxext / scipy / special.py _float_type   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / scipy / special.py _castto   100% 2 0 0   100% 0 0   100%
src / bartz / jaxext / scipy / special.py _castto.newfunc   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / scipy / special.py gammainccinv   100% 5 0 0   100% 0 0   100%
src / bartz / jaxext / scipy / special.py ndtri   60% 5 2 0   50% 2 1   57%
src / bartz / jaxext / scipy / special.py _ndtri   100% 27 0 0   100% 0 0   100%
src / bartz / jaxext / scipy / special.py _ndtri._create_polynomial   100% 4 0 0   100% 2 0   100%
src / bartz / jaxext / scipy / special.py (no function)   100% 13 0 0   100% 0 0   100%
src / bartz / jaxext / scipy / stats.py invgamma.ppf   100% 1 0 0   100% 0 0   100%
src / bartz / jaxext / scipy / stats.py (no function)   100% 4 0 0   100% 0 0   100%
src / bartz / mcmcloop.py BurninTrace.from_state   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcloop.py MainTrace.from_state   100% 5 0 0   100% 2 0   100%
src / bartz / mcmcloop.py Callback.__call__   100% 0 0 1   100% 0 0   100%
src / bartz / mcmcloop.py run_mcmc   100% 12 0 0   100% 6 0   100%
src / bartz / mcmcloop.py _empty_trace   100% 7 0 0   100% 2 0   100%
src / bartz / mcmcloop.py _compute_i_skip   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcloop.py _run_mcmc_inner_loop   100% 5 0 0   100% 0 0   100%
src / bartz / mcmcloop.py _run_mcmc_inner_loop.loop_impl   91% 11 1 0   75% 4 1   87%
src / bartz / mcmcloop.py _run_mcmc_inner_loop.loop_noop   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcloop.py _run_mcmc_inner_loop.loop   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcloop.py _save_state_to_trace   100% 9 0 0   100% 0 0   100%
src / bartz / mcmcloop.py _set   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcloop.py _set.at_set   100% 6 0 0   100% 4 0   100%
src / bartz / mcmcloop.py make_default_callback   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcloop.py make_default_callback.asarray_or_none   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcloop.py print_callback   100% 9 0 0   100% 0 0   100%
src / bartz / mcmcloop.py print_callback.get_cond   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcloop.py print_callback.line_report_branch   100% 6 0 0   100% 4 0   100%
src / bartz / mcmcloop.py print_callback.just_dot_branch   100% 3 0 0   100% 2 0   100%
src / bartz / mcmcloop.py _convert_jax_arrays_in_args   100% 4 0 0   100% 0 0   100%
src / bartz / mcmcloop.py _convert_jax_arrays_in_args.convert_jax_arrays   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcloop.py _convert_jax_arrays_in_args.convert_jax_arrays.convert_jax_array   60% 5 2 0   50% 4 2   56%
src / bartz / mcmcloop.py _convert_jax_arrays_in_args.new_func   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcloop.py _print_report   79% 14 3 0   62% 8 1   73%
src / bartz / mcmcloop.py TreesTrace.from_dataclass   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcloop.py evaluate_trace   100% 25 0 0   100% 2 0   100%
src / bartz / mcmcloop.py compute_varcount   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcloop.py (no function)   100% 64 0 2   100% 0 0   100%
src / bartz / mcmcstep / __init__.py (no function)   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py propose_moves   100% 11 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py propose_grow_moves   100% 9 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py choose_leaf   100% 9 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py growable_leaves   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py categorical   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py choose_variable   100% 6 0 0   100% 4 0   100%
src / bartz / mcmcstep / _moves.py fully_used_variables   100% 5 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py ancestor_variables   100% 6 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py split_range   100% 10 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py randint_exclude   100% 6 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py _process_exclude   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py categorical_exclude   100% 5 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py choose_split   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py compute_partial_ratio   100% 6 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py propose_prune_moves   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py choose_leaf_parent   100% 11 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py randint_masked   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _moves.py (no function)   100% 49 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py field   100% 8 0 0   100% 4 0   100%
src / bartz / mcmcstep / _state.py chain_vmap_axes   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py data_vmap_axes   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py _find_metadata   100% 14 0 0   100% 8 0   100%
src / bartz / mcmcstep / _state.py _find_metadata.is_leaf   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py _find_metadata.get_axes   100% 3 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py Forest.num_chains   100% 3 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py _init_shape_shifting_parameters   89% 19 2 0   83% 6 1   88%
src / bartz / mcmcstep / _state.py _parse_p_nonterminal   100% 4 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py init   86% 28 4 0   75% 8 2   83%
src / bartz / mcmcstep / _state.py _get_blocked_vars   100% 5 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py _add_chains   100% 3 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py _parse_mesh   100% 10 0 0   100% 6 0   100%
src / bartz / mcmcstep / _state.py _parse_target_platform   100% 11 0 0   100% 6 0   100%
src / bartz / mcmcstep / _state.py _auto_axes   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py _shard_state   100% 5 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py _shard_state.shard_leaf   100% 9 0 0   100% 6 0   100%
src / bartz / mcmcstep / _state.py _all_none_or_not_none   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py _asarray_or_none   100% 3 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py _get_platform   0% 3 3 0   0% 2 0   0%
src / bartz / mcmcstep / _state.py _choose_suffstat_batch_size   68% 25 8 0   57% 14 2   64%
src / bartz / mcmcstep / _state.py _choose_suffstat_batch_size.final_round   100% 4 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py get_axis_size   100% 4 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py _get_k_n   100% 4 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py chol_with_gersh   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py _chol_with_gersh_impl   100% 7 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py _inv_via_chol_with_gersh   100% 4 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py get_num_chains   100% 5 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py _chain_axes_with_keys   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py _chain_axes_with_keys.axis_if_key   100% 3 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py _get_mc_out_axes   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py _find_mesh   88% 8 1 0   100% 0 0   88%
src / bartz / mcmcstep / _state.py _find_mesh.find_mesh   100% 2 0 0   100% 2 0   100%
src / bartz / mcmcstep / _state.py _split_all_keys   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py _split_all_keys.split_key   100% 5 0 0   100% 4 0   100%
src / bartz / mcmcstep / _state.py vmap_chains   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py vmap_chains.auto_vmapped_fun   100% 11 0 0   100% 4 0   100%
src / bartz / mcmcstep / _state.py vmap_chains.auto_vmapped_fun.wrapped_fun   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _state.py (no function)   100% 71 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py step   100% 15 0 0   100% 4 0   100%
src / bartz / mcmcstep / _step.py step_trees   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py accept_moves_and_sample_leaves   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py accept_moves_parallel_stage   100% 21 0 0   88% 8 1   97%
src / bartz / mcmcstep / _step.py apply_grow_to_indices   100% 6 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _compute_count_or_prec_trees   100% 15 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py compute_count_trees   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py compute_prec_trees   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py complete_ratio   100% 12 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py adapt_leaf_trees_to_grow_indices   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _logdet_from_chol   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _precompute_likelihood_terms_uv   100% 7 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _precompute_likelihood_terms_mv   100% 10 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _precompute_likelihood_terms_mv._term_from_chol   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py precompute_likelihood_terms   100% 4 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py _precompute_leaf_terms_uv   100% 5 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py _precompute_leaf_terms_mv   100% 15 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py precompute_leaf_terms   100% 3 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py accept_moves_sequential_stage   100% 6 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py accept_moves_sequential_stage.loop   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py accept_move_and_sample_leaves   100% 25 0 0   100% 6 0   100%
src / bartz / mcmcstep / _step.py sum_resid   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _scatter_add   100% 11 0 0   100% 4 0   100%
src / bartz / mcmcstep / _step.py _get_shard_map_patch_kwargs   100% 3 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py _scatter_add_impl   100% 9 0 0   100% 4 0   100%
src / bartz / mcmcstep / _step.py _compute_likelihood_ratio_uv   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _compute_likelihood_ratio_mv   100% 6 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _compute_likelihood_ratio_mv._quadratic_form   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py compute_likelihood_ratio   100% 4 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py accept_moves_final_stage   100% 1 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py apply_moves_to_leaf_indices   100% 4 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py apply_moves_to_split_trees   100% 2 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _sample_wishart_bartlett   100% 10 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _step_error_cov_inv_uv   100% 9 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py _step_error_cov_inv_mv   100% 5 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py step_error_cov_inv   100% 4 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py step_z   100% 5 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py step_s   100% 6 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py step_theta   100% 11 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py _log_p_lamda   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py step_sparse   100% 3 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py _step_sparse   100% 5 0 0   100% 2 0   100%
src / bartz / mcmcstep / _step.py step_config   100% 3 0 0   100% 0 0   100%
src / bartz / mcmcstep / _step.py (no function)   100% 108 0 0   100% 0 0   100%
src / bartz / prepcovars.py parse_xinfo   100% 6 0 0   100% 0 0   100%
src / bartz / prepcovars.py quantilized_splits_from_matrix   100% 7 0 0   100% 2 0   100%
src / bartz / prepcovars.py quantilized_splits_from_matrix.quantilize   100% 1 0 0   100% 0 0   100%
src / bartz / prepcovars.py _quantilized_splits_from_matrix   100% 16 0 0   100% 4 0   100%
src / bartz / prepcovars.py _huge_value   100% 3 0 0   100% 2 0   100%
src / bartz / prepcovars.py _ensure_unsigned   100% 1 0 0   100% 0 0   100%
src / bartz / prepcovars.py _signed_to_unsigned   47% 15 8 0   40% 10 4   44%
src / bartz / prepcovars.py uniform_splits_from_matrix   100% 6 0 0   100% 0 0   100%
src / bartz / prepcovars.py bin_predictors   100% 4 0 0   100% 0 0   100%
src / bartz / prepcovars.py bin_predictors.bin_predictors   100% 2 0 0   100% 0 0   100%
src / bartz / prepcovars.py (no function)   100% 17 0 0   100% 0 0   100%
Total     95% 2525 132 10   84% 514 51   93%

No items found using the specified filter.