Coverage report: 32%

Files Functions Classes

coverage.py v7.13.5, created at 2026-05-05 11:22 +0200

File function   statements missing excluded   coverage
core / __init__.py (no function)   8 0 0   100%
core / attention.py Attention.__init__   40 3 0   92%
core / attention.py Attention._compute_angle   5 0 0   100%
core / attention.py Attention._apply_rope_grouped   7 0 0   100%
core / attention.py Attention._apply_rope_thd   8 0 0   100%
core / attention.py Attention._compute_cache   20 0 0   100%
core / attention.py Attention.reshape_head   7 0 0   100%
core / attention.py Attention.reshape_head._reshaping   1 0 0   100%
core / attention.py Attention.reshape_head._permute   1 0 0   100%
core / attention.py Attention._compute_attention   8 0 0   100%
core / attention.py Attention.__call__   85 10 0   88%
core / attention.py (no function)   16 0 0   100%
core / block.py RMSNorm.__init__   2 0 0   100%
core / block.py RMSNorm.__call__   2 0 0   100%
core / block.py _build_norm   3 0 0   100%
core / block.py Block.__init__   7 0 0   100%
core / block.py Block.__call__   5 0 0   100%
core / block.py (no function)   13 0 0   100%
core / config.py Config.__post_init__   20 3 0   85%
core / config.py Config.__repr__   3 0 0   100%
core / config.py Config.to_dict   1 0 0   100%
core / config.py Config.from_dict   2 0 0   100%
core / config.py Config.from_yaml   9 9 0   0%
core / config.py Config.save_yaml   2 0 0   100%
core / config.py (no function)   73 0 0   100%
core / generation.py _greedy_decode   1 1 0   0%
core / generation.py _sampling_decode   1 0 0   100%
core / generation.py decode   4 1 0   75%
core / generation.py _generate_toks   14 2 0   86%
core / generation.py _generate_toks.__apply_top_k   7 7 0   0%
core / generation.py _generate_toks.__apply_top_k.sample_from_top_k   2 2 0   0%
core / generation.py _generate_toks.__apply_top_p   7 7 0   0%
core / generation.py _generate_toks.__apply_top_p.sample_from_top_p   6 6 0   0%
core / generation.py _generate_toks.generate_with_kv_cache   4 0 0   100%
core / generation.py _generate_toks.prefill_or_no_cache   4 0 0   100%
core / generation.py _generate_toks._get_tok_id   17 3 0   82%
core / generation.py _generate_toks.prefill_no_cache   5 5 0   0%
core / generation.py generate   13 3 0   77%
core / generation.py (no function)   11 0 0   100%
core / lora.py LoRALinear.__init__   7 0 0   100%
core / lora.py LoRALinear.__call__   5 0 0   100%
core / lora.py LoRALinear.merge_weights   1 0 0   100%
core / lora.py (no function)   11 0 0   100%
core / mlp.py Activation.__init__   1 1 0   0%
core / mlp.py Activation.__call__   2 2 0   0%
core / mlp.py Swiglu.__call__   2 0 0   100%
core / mlp.py MLP.__init__   9 0 0   100%
core / mlp.py MLP.__call__   4 0 0   100%
core / mlp.py (no function)   13 0 0   100%
core / model.py Transformer.__init__   28 11 0   61%
core / model.py Transformer.__call__   24 3 0   88%
core / model.py Transformer.__call__.block_fn   1 0 0   100%
core / model.py Transformer.__call__._apply_block   1 0 0   100%
core / model.py Transformer.from_pretrained   21 21 0   0%
core / model.py (no function)   12 0 0   100%
core / moe.py MoE.__init__   4 0 0   100%
core / moe.py MoE.__call__   15 0 0   100%
core / moe.py (no function)   8 0 0   100%
core / old_model.py Attention.__init__   21 21 0   0%
core / old_model.py Attention.__init__.__compute_angle   4 4 0   0%
core / old_model.py Attention.__apply_rotation   8 8 0   0%
core / old_model.py Attention.__call__   39 39 0   0%
core / old_model.py Activation.__init__   1 1 0   0%
core / old_model.py Activation.__call__   2 2 0   0%
core / old_model.py Swiglu.__call__   2 2 0   0%
core / old_model.py MLP.__init__   7 7 0   0%
core / old_model.py MLP.__call__   4 4 0   0%
core / old_model.py MoE.__init__   4 4 0   0%
core / old_model.py MoE.__call__   15 15 0   0%
core / old_model.py Block.__init__   7 7 0   0%
core / old_model.py Block.__call__   5 5 0   0%
core / old_model.py Transformer.__init__   27 27 0   0%
core / old_model.py Transformer.__call__   22 22 0   0%
core / old_model.py Transformer.__call__.block_fn   1 1 0   0%
core / old_model.py (no function)   26 26 0   0%
core / output.py (no function)   10 0 0   100%
core / sharding.py make_mesh   4 1 0   75%
core / sharding.py replicate   2 0 0   100%
core / sharding.py shard_batch   2 0 0   100%
core / sharding.py num_devices   1 0 0   100%
core / sharding.py (no function)   11 0 0   100%
dantinox / __init__.py (no function)   15 2 0   87%
dantinox / bench.py _decode_step   1 1 0   0%
dantinox / bench.py _prefill_step   1 1 0   0%
dantinox / bench.py _load_config   13 13 0   0%
dantinox / bench.py _detect_vocab   9 9 0   0%
dantinox / bench.py _detect_vocab._get   6 6 0   0%
dantinox / bench.py _detect_vocab._unwrap   5 5 0   0%
dantinox / bench.py _load_model   11 11 0   0%
dantinox / bench.py _attn_type   5 5 0   0%
dantinox / bench.py _theoretical_cache_mb   5 5 0   0%
dantinox / bench.py _val_loss   9 9 0   0%
dantinox / bench.py _xla_costs   10 10 0   0%
dantinox / bench.py benchmark_run   62 62 0   0%
dantinox / bench.py benchmark_run._safe_div   1 1 0   0%
dantinox / bench.py benchmark_run._safe_round   1 1 0   0%
dantinox / bench.py BenchmarkRunner.__init__   5 5 0   0%
dantinox / bench.py BenchmarkRunner.__repr__   1 1 0   0%
dantinox / bench.py BenchmarkRunner.run   20 20 0   0%
dantinox / bench.py (no function)   39 0 0   100%
dantinox / cli.py _init_jax_cache   4 4 0   0%
dantinox / cli.py _add_config_overrides   4 4 0   0%
dantinox / cli.py _apply_overrides   5 5 0   0%
dantinox / cli.py _cmd_train   6 6 0   0%
dantinox / cli.py _cmd_generate   27 27 0   0%
dantinox / cli.py _cmd_sweep   13 13 0   0%
dantinox / cli.py _cmd_sweep._agent_fn   10 10 0   0%
dantinox / cli.py _cmd_find_lr   23 23 0   0%
dantinox / cli.py _cmd_push   3 3 0   0%
dantinox / cli.py _cmd_pull   3 3 0   0%
dantinox / cli.py _cmd_plot   6 6 0   0%
dantinox / cli.py _cmd_benchmark   8 8 0   0%
dantinox / cli.py _cmd_infbench   35 35 0   0%
dantinox / cli.py _build_parser   75 75 0   0%
dantinox / cli.py main   5 5 0   0%
dantinox / cli.py (no function)   24 24 0   0%
dantinox / exceptions.py (no function)   6 0 0   100%
dantinox / generator.py _stream_prefill   2 2 0   0%
dantinox / generator.py _stream_decode   2 2 0   0%
dantinox / generator.py _sample_logit   22 22 0   0%
dantinox / generator.py _load_checkpoint   53 27 0   49%
dantinox / generator.py Generator.__init__   5 0 0   100%
dantinox / generator.py Generator.__repr__   2 0 0   100%
dantinox / generator.py Generator._bpe_fix   4 2 0   50%
dantinox / generator.py Generator.generate   6 0 0   100%
dantinox / generator.py Generator.generate_batch   15 15 0   0%
dantinox / generator.py Generator.stream   18 18 0   0%
dantinox / generator.py (no function)   30 0 0   100%
dantinox / hub.py resolve_checkpoint   3 1 0   67%
dantinox / hub.py push   11 11 0   0%
dantinox / hub.py pull   7 7 0   0%
dantinox / hub.py (no function)   8 0 0   100%
dantinox / plots / __init__.py (no function)   0 0 0   100%
dantinox / plots / plot_3d.py _save   5 5 0   0%
dantinox / plots / plot_3d.py _ax3   4 4 0   0%
dantinox / plots / plot_3d.py load   7 7 0   0%
dantinox / plots / plot_3d.py _bpt_fn   10 10 0   0%
dantinox / plots / plot_3d.py _decode_flops_m   11 11 0   0%
dantinox / plots / plot_3d.py fig1_cache_surface   53 53 0   0%
dantinox / plots / plot_3d.py _set_cache_axes   11 11 0   0%
dantinox / plots / plot_3d.py fig2_quality_cube   21 21 0   0%
dantinox / plots / plot_3d.py fig3_efficiency_cube   24 24 0   0%
dantinox / plots / plot_3d.py fig4_serving_surface   46 46 0   0%
dantinox / plots / plot_3d.py (no function)   40 40 0   0%
dantinox / plots / plot_3d_dkv.py _save   5 5 0   0%
dantinox / plots / plot_3d_dkv.py _ax3   4 4 0   0%
dantinox / plots / plot_3d_dkv.py load   7 7 0   0%
dantinox / plots / plot_3d_dkv.py fig5_dkv_cache_seqlen   58 58 0   0%
dantinox / plots / plot_3d_dkv.py fig6_kv_decoupling   48 48 0   0%
dantinox / plots / plot_3d_dkv.py fig7_mla_quality   50 50 0   0%
dantinox / plots / plot_3d_dkv.py fig8_dkv_numblocks   40 40 0   0%
dantinox / plots / plot_3d_dkv.py (no function)   37 37 0   0%
dantinox / plots / plot_insights.py _save   5 5 0   0%
dantinox / plots / plot_insights.py load   5 5 0   0%
dantinox / plots / plot_insights.py _pareto_front   7 7 0   0%
dantinox / plots / plot_insights.py fig1_pareto   48 48 0   0%
dantinox / plots / plot_insights.py fig2_serving   64 64 0   0%
dantinox / plots / plot_insights.py fig3_mla_dial   49 49 0   0%
dantinox / plots / plot_insights.py (no function)   26 26 0   0%
dantinox / plots / plot_perf.py _save   5 5 0   0%
dantinox / plots / plot_perf.py load   7 7 0   0%
dantinox / plots / plot_perf.py load_batch   7 7 0   0%
dantinox / plots / plot_perf.py _decode_flops   16 16 0   0%
dantinox / plots / plot_perf.py _prefill_flops   2 2 0   0%
dantinox / plots / plot_perf.py fig1_cache_breakdown   46 46 0   0%
dantinox / plots / plot_perf.py fig2_seqlen_throughput   40 40 0   0%
dantinox / plots / plot_perf.py fig3_flops_vs_cache   36 36 0   0%
dantinox / plots / plot_perf.py fig4_batch_throughput   29 29 0   0%
dantinox / plots / plot_perf.py fig4_missing   9 9 0   0%
dantinox / plots / plot_perf.py fig5_prefill   49 49 0   0%
dantinox / plots / plot_perf.py (no function)   47 47 0   0%
dantinox / plotting.py _run_group   35 35 0   0%
dantinox / plotting.py Plotter.__init__   3 3 0   0%
dantinox / plotting.py Plotter.__repr__   1 1 0   0%
dantinox / plotting.py Plotter.run   20 20 0   0%
dantinox / plotting.py (no function)   14 0 0   100%
dantinox / trainer.py _build_schedule   21 0 0   100%
dantinox / trainer.py _build_optimizer   13 6 0   54%
dantinox / trainer.py _load_text   23 17 0   26%
dantinox / trainer.py _format_text   3 0 0   100%
dantinox / trainer.py _model_summary   7 0 0   100%
dantinox / trainer.py _cast_params   2 2 0   0%
dantinox / trainer.py _save_weights   3 0 0   100%
dantinox / trainer.py Trainer.__init__   1 0 0   100%
dantinox / trainer.py Trainer.__repr__   1 1 0   0%
dantinox / trainer.py Trainer.fit   129 25 0   81%
dantinox / trainer.py Trainer.fit.train_step   15 0 0   100%
dantinox / trainer.py Trainer.fit.train_step._loss   5 1 0   80%
dantinox / trainer.py Trainer.fit.eval_step   5 1 0   80%
dantinox / trainer.py Trainer.fit.estimate_loss   12 0 0   100%
dantinox / trainer.py Trainer.find_lr   49 49 0   0%
dantinox / trainer.py Trainer.find_lr._lr_fn   1 1 0   0%
dantinox / trainer.py Trainer.find_lr._step   4 4 0   0%
dantinox / trainer.py Trainer.find_lr._step.loss_fn   2 2 0   0%
dantinox / trainer.py (no function)   34 0 0   100%
Total     2883 1967 0   32%

No items found using the specified filter.