fix bug remove

stanfordnmbl · Jul 30, 2024 · e0bf5b3 · e0bf5b3
1 parent a1d7678
commit e0bf5b3
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 12 deletions.
diff --git a/gait_analysis/function/handler.py b/gait_analysis/function/handler.py
@@ -209,10 +209,14 @@ def handler(event, context):
     # Available options for line curve chart.
     y_axes = list(colNames)
     y_axes.remove('time')
-    y_axes.remove('knee_angle_r_beta')
-    y_axes.remove('knee_angle_l_beta')
-    y_axes.remove('mtp_angle_r')
-    y_axes.remove('mtp_angle_l')
+    if 'knee_angle_r_beta' in y_axes:
+        y_axes.remove('knee_angle_r_beta')
+    if 'knee_angle_l_beta' in y_axes:
+        y_axes.remove('knee_angle_l_beta')
+    if 'mtp_angle_r' in y_axes:
+        y_axes.remove('mtp_angle_r')
+    if 'mtp_angle_l' in y_axes:
+        y_axes.remove('mtp_angle_l')
 
     # Create results dictionnary.
     results = {

diff --git a/squat_analysis/function/handler.py b/squat_analysis/function/handler.py
@@ -141,10 +141,14 @@ def handler(event, context):
     # Available options for line curve chart.
     y_axes = list(colNames)
     y_axes.remove('time')
-    y_axes.remove('knee_angle_r_beta')
-    y_axes.remove('knee_angle_l_beta')
-    y_axes.remove('mtp_angle_r')
-    y_axes.remove('mtp_angle_l')
+    if 'knee_angle_r_beta' in y_axes:
+        y_axes.remove('knee_angle_r_beta')
+    if 'knee_angle_l_beta' in y_axes:
+        y_axes.remove('knee_angle_l_beta')
+    if 'mtp_angle_r' in y_axes:
+        y_axes.remove('mtp_angle_r')
+    if 'mtp_angle_l' in y_axes:
+        y_axes.remove('mtp_angle_l')
 
     # Create results dictionary.
     results = {

diff --git a/treadmill_gait_analysis/function/handler.py b/treadmill_gait_analysis/function/handler.py
@@ -208,10 +208,14 @@ def handler(event, context):
     # Available options for line curve chart.
     y_axes = list(colNames)
     y_axes.remove('time')
-    y_axes.remove('knee_angle_r_beta')
-    y_axes.remove('knee_angle_l_beta')
-    y_axes.remove('mtp_angle_r')
-    y_axes.remove('mtp_angle_l')
+    if 'knee_angle_r_beta' in y_axes:
+        y_axes.remove('knee_angle_r_beta')
+    if 'knee_angle_l_beta' in y_axes:
+        y_axes.remove('knee_angle_l_beta')
+    if 'mtp_angle_r' in y_axes:
+        y_axes.remove('mtp_angle_r')
+    if 'mtp_angle_l' in y_axes:
+        y_axes.remove('mtp_angle_l')
 
     # Create results dictionnary.
     results = {