bella-companion 0.0.7__py3-none-any.whl → 0.0.8__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of bella-companion might be problematic. Click here for more details.

Files changed (26) hide show
  1. bella_companion/cli.py +14 -4
  2. bella_companion/fbd_empirical/__init__.py +2 -1
  3. bella_companion/fbd_empirical/notbooks.ipynb +85 -274
  4. bella_companion/fbd_empirical/run_beast.py +3 -3
  5. bella_companion/fbd_empirical/summarize_logs.py +15 -32
  6. bella_companion/simulations/__init__.py +8 -1
  7. bella_companion/simulations/figures/epi_skyline_results.py +2 -2
  8. bella_companion/simulations/figures/explain/pdp.py +21 -27
  9. bella_companion/simulations/figures/explain/shap.py +1 -1
  10. bella_companion/simulations/figures/fbd_2traits_results.py +1 -1
  11. bella_companion/simulations/figures/fbd_no_traits_results.py +2 -2
  12. bella_companion/simulations/figures/scenarios.py +2 -2
  13. bella_companion/simulations/generate_data.py +4 -2
  14. bella_companion/simulations/metrics.py +62 -0
  15. bella_companion/simulations/scenarios/epi_multitype.py +1 -1
  16. bella_companion/simulations/scenarios/fbd_2traits.py +1 -4
  17. bella_companion/simulations/summarize_logs.py +3 -4
  18. bella_companion/utils/__init__.py +7 -11
  19. bella_companion/utils/beast.py +2 -2
  20. {bella_companion-0.0.7.dist-info → bella_companion-0.0.8.dist-info}/METADATA +2 -2
  21. bella_companion-0.0.8.dist-info/RECORD +42 -0
  22. bella_companion/fbd_empirical/figure.py +0 -37
  23. bella_companion-0.0.7.dist-info/RECORD +0 -42
  24. /bella_companion/{utils/plots.py → simulations/figures/utils.py} +0 -0
  25. {bella_companion-0.0.7.dist-info → bella_companion-0.0.8.dist-info}/WHEEL +0 -0
  26. {bella_companion-0.0.7.dist-info → bella_companion-0.0.8.dist-info}/entry_points.txt +0 -0
bella_companion/cli.py CHANGED
@@ -5,9 +5,10 @@ from pathlib import Path
5
5
  from dotenv import load_dotenv
6
6
 
7
7
  from bella_companion.fbd_empirical import run_beast as run_fbd_empirical
8
- from bella_companion.simulations import generate_data, generate_figures
8
+ from bella_companion.fbd_empirical import summarize_logs as summarize_fbd_empirical
9
+ from bella_companion.simulations import generate_data, generate_figures, print_metrics
9
10
  from bella_companion.simulations import run_beast as run_simulations
10
- from bella_companion.simulations import summarize_logs
11
+ from bella_companion.simulations import summarize_logs as summarize_simulations
11
12
 
12
13
 
13
14
  def main():
@@ -29,8 +30,12 @@ def main():
29
30
  ).set_defaults(func=run_simulations)
30
31
 
31
32
  subparsers.add_parser(
32
- "sim-summary", help="Summarize BEAST2 log outputs for simulations."
33
- ).set_defaults(func=summarize_logs)
33
+ "sim-summarize", help="Summarize BEAST2 log outputs for simulations."
34
+ ).set_defaults(func=summarize_simulations)
35
+
36
+ subparsers.add_parser(
37
+ "sim-metrics", help="Compute and print metrics from simulation results."
38
+ ).set_defaults(func=print_metrics)
34
39
 
35
40
  subparsers.add_parser(
36
41
  "sim-figures", help="Generate plots and figures from simulation results."
@@ -40,5 +45,10 @@ def main():
40
45
  "fbd-empirical-run", help="Run BEAST2 analyses on empirical FBD datasets."
41
46
  ).set_defaults(func=run_fbd_empirical)
42
47
 
48
+ subparsers.add_parser(
49
+ "fbd-empirical-summarize",
50
+ help="Summarize BEAST2 log outputs for empirical FBD datasets.",
51
+ ).set_defaults(func=summarize_fbd_empirical)
52
+
43
53
  args = parser.parse_args()
44
54
  args.func()
@@ -1,3 +1,4 @@
1
1
  from bella_companion.fbd_empirical.run_beast import run_beast
2
+ from bella_companion.fbd_empirical.summarize_logs import summarize_logs
2
3
 
3
- __all__ = ["run_beast"]
4
+ __all__ = ["run_beast", "summarize_logs"]