fix custom graphs not working

team4099 · Feb 14, 2024 · eaa0898 · eaa0898
1 parent 6bca543
commit eaa0898
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/page_managers/custom_graphs_manager.py b/src/page_managers/custom_graphs_manager.py
@@ -40,7 +40,7 @@ def generate_input_section(self) -> list[list, list, Callable, str]:
         names_to_methods = {
             name.replace("_", " ").capitalize(): method
             for name, method in inspect.getmembers(self.calculated_stats, predicate=inspect.ismethod)
-            if not name.startswith("__") and "(ignore)" not in method.__doc__
+            if method.__doc__ and not name.startswith("__") and "(ignore)" not in method.__doc__
         }
 
         st.write("#### 📈 Data to Display")

diff --git a/src/utils/calculated_stats.py b/src/utils/calculated_stats.py
@@ -93,7 +93,7 @@ def average_cycles(self, team_number: int, mode: str = None) -> float:
         :param mode: The mode to calculate said cycles for (Auto/Teleop)
         :return: A float representing the average cycles for said team in the mode specified.
         """
-        if mode:
+        if mode is not None:
             return self.cycles_by_match(team_number, mode).mean()
         else:
             return (self.cycles_by_match(team_number, Queries.AUTO) + self.cycles_by_match(team_number, Queries.TELEOP)).mean()