set in for both cnmf and cnmfe in addition to eval kwargs

kushalkolar · Nov 12, 2021 · 8d43d80 · 8d43d80
1 parent 16fca7a
commit 8d43d80
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/mesmerize/viewer/modules/cnmf.py b/mesmerize/viewer/modules/cnmf.py
@@ -78,6 +78,7 @@ def get_params(self, *args, group_params: bool = False) -> dict:
                 'tsub': self.ui.spinBox_tsub.value(),
                 'method_init': self.ui.comboBox_method_init.currentText(),
                 'border_pix': bord_px,
+                'fr': self.vi.viewer.workEnv.imgdata.meta['fps']
             }
 
         # Any additional cnmf kwargs set in the text entry

diff --git a/mesmerize/viewer/modules/cnmfe.py b/mesmerize/viewer/modules/cnmfe.py
@@ -96,7 +96,8 @@ def get_params(self, item_type: str, group_params: bool = False) -> dict:
                 'low_rank_background': self.ui.checkBox_low_rank_background.isChecked(),
                 'ring_size_factor': self.ui.doubleSpinBox_ring_size_factor.value(),
                 'update_background_components': True,
-                'del_duplicates': True
+                'del_duplicates': True,
+                'fr': self.vi.viewer.workEnv.imgdata.meta['fps']
             }
 
         # Any additional CNMFE kwargs set in the text entry