From 7a6e7d42164575f05e17088a29b85c9f1da2db0b Mon Sep 17 00:00:00 2001
From: Christos Paschalidis <80837632+Paschas@users.noreply.github.com>
Date: Fri, 26 Jan 2024 14:59:46 +0200
Subject: [PATCH] Update main.py

first try commiting
---
 utils2p/main.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/utils2p/main.py b/utils2p/main.py
index c0f7baa..7f6f0c2 100644
--- a/utils2p/main.py
+++ b/utils2p/main.py
@@ -371,12 +371,12 @@ def get_frame_rate(self):
         Returns the frame rate for a given experiment metadata.
         When the frame rate is calculated flyback frames and
         steps in z are not considered frames.
-
+    
         Returns
         -------
         frame_rate : float
             Frame rate of the experiment.
-
+    
         Examples
         --------
         >>> import utils2p
@@ -387,9 +387,12 @@ def get_frame_rate(self):
         frame_rate_without_flybacks = float(
             self.get_metadata_value("LSM", "frameRate"))
         flyback_frames = self.get_n_flyback_frames()
-        number_of_slices = self.get_n_z()
+        
+        enable_z_stage = float(self.get_metadata_value("ZStage", "enable"))
+        number_of_slices = self.get_n_z() if enable_z_stage else 0
+        
         return frame_rate_without_flybacks / (flyback_frames +
-                                              number_of_slices)
+                                                number_of_slices)
 
     def get_width(self):
         """