in case predict_ndarray_fb doesn't find picks, it return -999.03125

geo-stack · Jan 16, 2025 · 244efbb · 244efbb
1 parent bfe109e
commit 244efbb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/first_break_picking/train_eval/predict.py b/first_break_picking/train_eval/predict.py
@@ -172,7 +172,10 @@ def predict_ndarray_fb(self,
                     data_info=data_info,
                     case_specific_parameters=self.case_specific_parameters
                 )
-        return np.array(predicted_pick * self.dt)
+        if predicted_pick is None:
+            return -999.03125 * np.ones(shot.shape[1])
+        else:
+            return np.array(predicted_pick * self.dt)
 
     def predict(self, path_data: str):
         """Predict FB