buf fix since no second ROI with stim crosscorr

kushalkolar · Aug 14, 2020 · 7711a8d · 7711a8d
1 parent f8564ba
commit 7711a8d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mesmerize/plotting/widgets/cross_correlation/widget.py b/mesmerize/plotting/widgets/cross_correlation/widget.py
@@ -204,8 +204,9 @@ def add_second_roi_to_datapoint_tracer(self, r: pd.Series):
         elif roi_state['roi_type'] == 'ManualROI':
             self.roi_2 = ManualROI.from_state(curve_plot_item=None, view_box=self.datapoint_tracer.view, state=roi_state)
 
-        self.roi_2.get_roi_graphics_object().setPen(mkColor('c'))
-        self.roi_2.add_to_viewer()
+        if self.roi_2 is not None:
+            self.roi_2.get_roi_graphics_object().setPen(mkColor('c'))
+            self.roi_2.add_to_viewer()
 
     def _get_xticks_linspace(self, ncc) -> np.ndarray:
             m = ncc.size