fix issue (#434)

preserve attributes when splitting data
OHDSI · Mar 27, 2024 · 2b5f007 · 2b5f007
1 parent 11122d4
commit 2b5f007
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,6 +4,8 @@
 *-Ex.R
 # R data files from past sessions
 .Rdata
+# R environ
+.Renviron
 # RStudio files
 .Rproj.user/
 .Rproj.user
@@ -20,4 +22,4 @@ standalone/build/*
 /plpmodels/*
 /python_models/*
 /mycache/*
-/inst/shiny/DiagnosticsExplorer/rsconnect/*
+/inst/shiny/DiagnosticsExplorer/rsconnect/*
diff --git a/R/DataSplitting.R b/R/DataSplitting.R
@@ -197,10 +197,6 @@ splitData <- function(plpData = plpData,
       populationSize = nrow(trainData$labels)
       )
 
-    # add pop size to covariateData as used in tidyCovariates
-    attr(trainData$covariateData, "metaData") <- list(populationSize = nrow(trainData$labels))
-    class(trainData$covariateData) <- "CovariateData"
-
     testId <- splitId[splitId$index<0,]
     testData <- list()
     class(testData) <- 'plpData'
@@ -216,7 +212,6 @@ splitData <- function(plpData = plpData,
         data.frame(rowId = testId$rowId), 
         sizeN = 10000000)
     }
-    class(testData$covariateData) <- "CovariateData"
 
     result <- list(
       Train =  trainData,