conditional return on exception

AllenInstitute · Dec 11, 2024 · 0a437f7 · 0a437f7
1 parent c145492
commit 0a437f7
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/dynamic_routing_analysis/decoding_utils.py b/src/dynamic_routing_analysis/decoding_utils.py
@@ -1262,7 +1262,8 @@ def decode_context_with_linear_shift(session=None,params=None,trials=None,units=
         print(tb_str)
         logger.debug(f'error in session {session_id}')
         logger.debug(tb_str)
-        return None
+        if return_results:
+            return None
 
 
 def concat_decoder_results(files,savepath=None,return_table=True,single_session=False,use_zarr=False):
@@ -1515,7 +1516,8 @@ def concat_decoder_results(files,savepath=None,return_table=True,single_session=
         print(tb_str)
         logger.debug(f'error with decoding results summary')
         logger.debug(tb_str)
-        return None
+        if return_table:
+            return None
 
 
 def compute_significant_decoding_by_area(all_decoder_results):
@@ -2474,7 +2476,8 @@ def concat_trialwise_decoder_results(files,savepath=None,return_table=False,n_un
         print(tb_str)
         logger.debug(f'error with trialwise decoding results summary')
         logger.debug(tb_str)
-        return None
+        if return_table:
+            return None,None,None,None,None
 
 
 def concat_decoder_summary_tables(dir,savepath):