DOR-736 Add test for handling very short reads that are entirely trim…

…med by the ScalerNode and fix the crash they cause.
nanoporetech · Jun 3, 2024 · 03d94f3 · 03d94f3
1 parent f9beb39
commit 03d94f3
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/dorado/read_pipeline/ScalerNode.cpp b/dorado/read_pipeline/ScalerNode.cpp
@@ -141,7 +141,7 @@ void ScalerNode::input_thread_fn() {
         int trim_start = 0;
         if (is_rna_model) {
             trim_start = determine_rna_adapter_pos(*read, m_model_type);
-            if (m_trim_rna_adapter) {
+            if (m_trim_rna_adapter && trim_start < read->read_common.get_raw_data_samples()) {
                 read->read_common.raw_data =
                         read->read_common.raw_data.index({Slice(trim_start, at::indexing::None)});
                 read->read_common.rna_adapter_end_signal_pos = 0;
@@ -241,8 +241,12 @@ void ScalerNode::input_thread_fn() {
                         utils::DEFAULT_TRIM_MIN_ELEMENTS);
             }
 
-            read->read_common.raw_data =
-                    read->read_common.raw_data.index({Slice(trim_start, at::indexing::None)});
+            if (trim_start < read->read_common.get_raw_data_samples()) {
+                read->read_common.raw_data =
+                        read->read_common.raw_data.index({Slice(trim_start, at::indexing::None)});
+            } else {
+                trim_start = 0;
+            }
         }
 
         read->read_common.num_trimmed_samples = trim_start;

diff --git a/tests/data/pod5/degenerate/trimming_bomb.pod5 b/tests/data/pod5/degenerate/trimming_bomb.pod5
diff --git a/tests/test_simple_basecaller_execution.sh b/tests/test_simple_basecaller_execution.sh
@@ -78,6 +78,9 @@ if $dorado_bin basecaller $model_5k_v43 $data_dir/duplex/pod5 --modified-bases 5
 fi
 set -e
 
+# Check that dorado handles degenerate reads without crashing
+$dorado_bin basecaller $model_5k_v43 $data_dir/pod5/degenerate --skip-model-compatibility-check > $output_dir/error_condition.fq
+
 echo dorado summary test stage
 $dorado_bin summary $output_dir/calls.bam