Refactor num_entries as float

broadinstitute · Jan 11, 2024 · a3fe0f7 · a3fe0f7
1 parent f08a683
commit a3fe0f7
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/wdl/BenchmarkGqFilter.wdl b/wdl/BenchmarkGqFilter.wdl
@@ -153,7 +153,7 @@ task BenchmarkFilter {
         String benchmark_figure_filename = "quality-benchmark.pdf"
         Array[String] benchmark_args = []
         String sv_utils_docker
-        Int num_entries
+        Float num_entries
         Float pickled_files_size
     }
 

diff --git a/wdl/PickleVcfProperties.wdl b/wdl/PickleVcfProperties.wdl
@@ -35,15 +35,15 @@ workflow PickleVcfProperties {
     output {
         Int num_records = GetVcfSize.num_records
         Int num_samples = GetVcfSize.num_samples
-        Int num_entries = GetVcfSize.num_entries
+        Float num_entries = GetVcfSize.num_entries
         File pickled_properties = ReadAndPickleProperties.pickled_properties
     }
 }
 
 task GetNeededMemGB {
     input {
         Array[String] wanted_properties
-        Int num_entries
+        Float num_entries
     }
 
     Float mem_gb = "1.5"

diff --git a/wdl/TrainGqRecalibrator.wdl b/wdl/TrainGqRecalibrator.wdl
@@ -48,7 +48,7 @@ task TrainGqRecalibratorTask {
         File? gq_recalibrator_model_file # can be passed to do extra rounds of training on existing model
         Array[String] train_args = []
         String gatk_docker
-        Int? num_entries
+        Float? num_entries
         Float mem_scale_vcf_size = 25.2
         Float mem_scale_num_entries = "3.7e-7"
         Float mem_gb_overhead = 1.5

diff --git a/wdl/Utils.wdl b/wdl/Utils.wdl
@@ -498,7 +498,7 @@ task GetVcfSize {
     output {
         Int num_records = read_int(num_records_file)
         Int num_samples = read_int(num_samples_file)
-        Int num_entries = num_records * num_samples
+        Float num_entries = float(num_records) * num_samples
     }
 }