From 06490d7ee110af70508b574cdf2fcfaa7dfc3d74 Mon Sep 17 00:00:00 2001 From: Oskar Laverny Date: Sat, 11 May 2024 12:15:35 +0200 Subject: [PATCH 1/2] add precompile statements --- Project.toml | 3 ++- src/NetSurvival.jl | 15 ++++++++++++++- 2 files changed, 16 insertions(+), 2 deletions(-) diff --git a/Project.toml b/Project.toml index 740fbed..bf338b9 100644 --- a/Project.toml +++ b/Project.toml @@ -8,6 +8,7 @@ CSV = "336ed68f-0bac-5ca0-87d4-7b16caf5d00b" DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0" Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f" LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e" +PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a" RateTables = "d40fb65e-c2ee-4113-9e14-cb96ca0acb32" StatsAPI = "82ae8749-77ed-4fe6-ae5f-f523153014b0" StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91" @@ -19,8 +20,8 @@ CSV = "0.10" DataFrames = "1" Distributions = "0.25" LinearAlgebra = "1.6" -RateTables = "0.1" RCall = "0.14" +RateTables = "0.1" StatsAPI = "1" StatsBase = "0.34" StatsModels = "0.7" diff --git a/src/NetSurvival.jl b/src/NetSurvival.jl index abead04..ea821a8 100644 --- a/src/NetSurvival.jl +++ b/src/NetSurvival.jl @@ -1,5 +1,5 @@ module NetSurvival - +using PrecompileTools: @setup_workload, @compile_workload # this is a small dependency using DataFrames using Distributions using LinearAlgebra @@ -32,4 +32,17 @@ export Surv, Strata export colrec export @formula +@setup_workload begin + # Putting some things in `@setup_workload` instead of `@compile_workload` can reduce the size of the + # precompile file and potentially make loading faster. + @compile_workload begin + fit(PoharPerme, @formula(Surv(time,status)~1), colrec, slopop) + fit(PoharPerme, @formula(Surv(time,status)~sex), colrec, slopop) + fit(GraffeoTest, @formula(Surv(time,status)~sex), colrec, slopop) + fit(GraffeoTest, @formula(Surv(time,status)~sex+Strata(stage)), colrec, slopop) + fit(CrudeMortality, @formula(Surv(time,status)~1), colrec, slopop) + end +end + + end \ No newline at end of file From 6881844871287977d94e49f62960bce984eee503 Mon Sep 17 00:00:00 2001 From: Oskar Laverny Date: Sat, 11 May 2024 12:17:32 +0200 Subject: [PATCH 2/2] add other methods ? --- src/NetSurvival.jl | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/NetSurvival.jl b/src/NetSurvival.jl index ea821a8..6d0e654 100644 --- a/src/NetSurvival.jl +++ b/src/NetSurvival.jl @@ -38,6 +38,9 @@ export @formula @compile_workload begin fit(PoharPerme, @formula(Surv(time,status)~1), colrec, slopop) fit(PoharPerme, @formula(Surv(time,status)~sex), colrec, slopop) + fit(EdererI, @formula(Surv(time,status)~1), colrec, slopop) + fit(EdererII, @formula(Surv(time,status)~1), colrec, slopop) + fit(Hakulinen, @formula(Surv(time,status)~1), colrec, slopop) fit(GraffeoTest, @formula(Surv(time,status)~sex), colrec, slopop) fit(GraffeoTest, @formula(Surv(time,status)~sex+Strata(stage)), colrec, slopop) fit(CrudeMortality, @formula(Surv(time,status)~1), colrec, slopop)