Fixed small test typo

mmcdermott · Nov 8, 2024 · 7c5d3bf · 7c5d3bf
1 parent cab7b32
commit 7c5d3bf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/utils.py b/tests/utils.py
@@ -405,6 +405,9 @@ def single_stage_tester(
     if df_check_kwargs is None:
         df_check_kwargs = {}
 
+    if stage_kwargs is None:
+        stage_kwargs = {}
+
     with input_dataset(input_files) as (input_dir, cohort_dir):
         for k, v in pipeline_kwargs.items():
             if type(v) is str and "{input_dir}" in v: