Cleanup dictionaries during unload (#233)

* cleanup dicts during unload * rename file
mlr-org · Dec 24, 2023 · 3dd9eac · 3dd9eac
1 parent d4485a1
commit 3dd9eac
Show file tree

Hide file tree

Showing 24 changed files with 102 additions and 1 deletion.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -65,3 +65,39 @@ LazyData: true
 NeedsCompilation: no
 Roxygen: list(markdown = TRUE, r6 = TRUE)
 RoxygenNote: 7.2.3
+Collate: 
+    'aaa.R'
+    'ResamplingRepeatedSpCVBlock.R'
+    'ResamplingRepeatedSpCVCoords.R'
+    'ResamplingRepeatedSpCVDisc.R'
+    'ResamplingRepeatedSpCVEnv.R'
+    'ResamplingRepeatedSpCVTiles.R'
+    'ResamplingRepeatedSpCVknndm.R'
+    'ResamplingRepeatedSptCVCstf.R'
+    'ResamplingSpCVBlock.R'
+    'ResamplingSpCVBuffer.R'
+    'ResamplingSpCVCoords.R'
+    'ResamplingSpCVDisc.R'
+    'ResamplingSpCVEnv.R'
+    'ResamplingSpCVKnndm.R'
+    'ResamplingSpCVTiles.R'
+    'ResamplingSptCVCstf.R'
+    'TaskClassifST.R'
+    'TaskRegrST.R'
+    'Task_classif_diplodia.R'
+    'Task_classif_ecuador.R'
+    'Task_regr_cookfarm_profiles.R'
+    'as_task_classif_st.R'
+    'as_task_regr_st.R'
+    'autoplot.R'
+    'autoplot_all_folds_dt.R'
+    'autoplot_all_folds_list.R'
+    'autoplot_multi_fold_dt.R'
+    'autoplot_multi_fold_list.R'
+    'autoplot_spcv_cstf.R'
+    'bibentries.R'
+    'helper.R'
+    'helper_DataBackend.R'
+    'helper_autoplot.R'
+    'reexports.R'
+    'zzz.R'
diff --git a/R/ResamplingRepeatedSpCVBlock.R b/R/ResamplingRepeatedSpCVBlock.R
@@ -262,3 +262,6 @@ ResamplingRepeatedSpCVBlock = R6Class("ResamplingRepeatedSpCVBlock",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["repeated_spcv_block"]] = ResamplingRepeatedSpCVBlock
diff --git a/R/ResamplingRepeatedSpCVCoords.R b/R/ResamplingRepeatedSpCVCoords.R
@@ -132,3 +132,6 @@ ResamplingRepeatedSpCVCoords = R6Class("ResamplingRepeatedSpCVCoords",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["repeated_spcv_coords"]] = ResamplingRepeatedSpCVCoords
diff --git a/R/ResamplingRepeatedSpCVDisc.R b/R/ResamplingRepeatedSpCVDisc.R
@@ -198,3 +198,6 @@ ResamplingRepeatedSpCVDisc = R6Class("ResamplingRepeatedSpCVDisc",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["repeated_spcv_disc"]] = ResamplingRepeatedSpCVDisc
diff --git a/R/ResamplingRepeatedSpCVEnv.R b/R/ResamplingRepeatedSpCVEnv.R
@@ -151,3 +151,6 @@ ResamplingRepeatedSpCVEnv = R6Class("ResamplingRepeatedSpCVEnv",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["repeated_spcv_env"]] = ResamplingRepeatedSpCVEnv
diff --git a/R/ResamplingRepeatedSpCVTiles.R b/R/ResamplingRepeatedSpCVTiles.R
@@ -364,3 +364,6 @@ ResamplingRepeatedSpCVTiles = R6Class("ResamplingRepeatedSpCVTiles",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["repeated_spcv_tiles"]] = ResamplingRepeatedSpCVTiles
diff --git a/R/ResamplingRepeatedSpCVknndm.R b/R/ResamplingRepeatedSpCVknndm.R
@@ -197,3 +197,6 @@ ResamplingRepeatedSpCVKnndm = R6Class("ResamplingRepeatedSpCVKnndm",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["repeated_spcv_knndm"]] = ResamplingRepeatedSpCVKnndm
diff --git a/R/ResamplingRepeatedSptCVCstf.R b/R/ResamplingRepeatedSptCVCstf.R
@@ -132,3 +132,6 @@ ResamplingRepeatedSptCVCstf = R6Class("ResamplingRepeatedSptCVCstf",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["repeated_sptcv_cstf"]] = ResamplingRepeatedSptCVCstf
diff --git a/R/ResamplingSpCVBlock.R b/R/ResamplingSpCVBlock.R
@@ -199,3 +199,6 @@ ResamplingSpCVBlock = R6Class("ResamplingSpCVBlock",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["spcv_block"]] = ResamplingSpCVBlock
diff --git a/R/ResamplingSpCVBuffer.R b/R/ResamplingSpCVBuffer.R
@@ -172,3 +172,6 @@ ResamplingSpCVBuffer = R6Class("ResamplingSpCVBuffer",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["spcv_buffer"]] = ResamplingSpCVBuffer
diff --git a/R/ResamplingSpCVCoords.R b/R/ResamplingSpCVCoords.R
@@ -97,3 +97,6 @@ ResamplingSpCVCoords = R6Class("ResamplingSpCVCoords",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["spcv_coords"]] = ResamplingSpCVCoords
diff --git a/R/ResamplingSpCVDisc.R b/R/ResamplingSpCVDisc.R
@@ -161,3 +161,6 @@ ResamplingSpCVDisc = R6Class("ResamplingSpCVDisc",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["spcv_disc"]] = ResamplingSpCVDisc
diff --git a/R/ResamplingSpCVEnv.R b/R/ResamplingSpCVEnv.R
@@ -127,3 +127,6 @@ ResamplingSpCVEnv = R6Class("ResamplingSpCVEnv",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["spcv_env"]] = ResamplingSpCVEnv
diff --git a/R/ResamplingSpCVknndm.R → R/ResamplingSpCVKnndm.R b/R/ResamplingSpCVknndm.R → R/ResamplingSpCVKnndm.R
@@ -179,3 +179,6 @@ ResamplingSpCVKnndm = R6Class("ResamplingSpCVKnndm",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["spcv_knndm"]] = ResamplingSpCVKnndm
diff --git a/R/ResamplingSpCVTiles.R b/R/ResamplingSpCVTiles.R
@@ -312,3 +312,6 @@ ResamplingSpCVTiles = R6Class("ResamplingSpCVTiles",
     }
   )
 )
+
+#' @include aaa.R
+resamplings[["spcv_tiles"]] = ResamplingSpCVTiles
diff --git a/R/ResamplingSptCVCstf.R b/R/ResamplingSptCVCstf.R
@@ -163,3 +163,6 @@ sample_cast = function(task, stratify = FALSE, folds) {
   test = if (length(space) && length(time)) pmap(list(test_space, test_time), function(x, y) intersect(x, y)) else if (length(space)) test_space else test_time
   list(train = train, test = test)
 }
+
+#' @include aaa.R
+resamplings[["sptcv_cstf"]] = ResamplingSptCVCstf
diff --git a/R/aaa.R b/R/aaa.R
@@ -0,0 +1 @@
+resamplings = list()
diff --git a/R/zzz.R b/R/zzz.R
@@ -123,6 +123,9 @@ register_mlr3 = function() { # nocov start
   hooks = getHook(event)
   pkgname = vapply(hooks, function(x) environment(x)$pkgname, NA_character_)
   setHook(event, hooks[pkgname != "mlr3spatiotempcv"], action = "replace")
+
+  mlr_resamplings = mlr3::mlr_resamplings
+  walk(names(resamplings), function(id) mlr_resamplings$remove(id))
 }
 
 leanify_package() # nocov end
diff --git a/man/as_task_regr_st.Rd b/man/as_task_regr_st.Rd
diff --git a/man/autoplot.ResamplingCV.Rd b/man/autoplot.ResamplingCV.Rd
diff --git a/man/mlr_resamplings_repeated_sptcv_cstf.Rd b/man/mlr_resamplings_repeated_sptcv_cstf.Rd
diff --git a/man/mlr_resamplings_spcv_knndm.Rd b/man/mlr_resamplings_spcv_knndm.Rd
diff --git a/man/mlr_resamplings_sptcv_cstf.Rd b/man/mlr_resamplings_sptcv_cstf.Rd
diff --git a/tests/testthat/test-unload.R b/tests/testthat/test-unload.R
@@ -0,0 +1,8 @@
+test_that("unloading leaves no trace", {
+  library(mlr3spatiotempcv)
+  n_resamplings = length(resamplings)
+  n_total = length(mlr_resamplings$keys())
+  unloadNamespace("mlr3spatiotempcv")
+  n_mlr = length(mlr_resamplings$keys())
+  expect_true(n_resamplings == n_total - n_mlr)
+})
-Original file line number
+Diff line change
@@ Expand Up @@
         }
       )
     )
+    #' @include aaa.R
+    resamplings[["repeated_spcv_block"]] = ResamplingRepeatedSpCVBlock