Update MABHandler.java

Make it return a selected arm correctly
RoboTutorLLC · Oct 21, 2024 · 9e2a5d3 · 9e2a5d3
1 parent e73395d
commit 9e2a5d3
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/app/src/main/java/cmu/xprize/robotutor/tutorengine/util/MABHandler.java b/app/src/main/java/cmu/xprize/robotutor/tutorengine/util/MABHandler.java
@@ -33,9 +33,14 @@ public class MABHandler {
     public static String getArm(String dataSource, IScope2 scope) {
         List<Arm> arms = getarms(dataSource, scope);
         Arm selectedArm = selectArm(arms);
-        Log.d(TAG, "getArm: list = " + arms);
-        Log.d(TAG, "getArm: selected = " + selectedArm);
-        return "";
+        // Ensure that selected arm is not null
+        if (selectedArm != null) {
+            Log.d(TAG, "getArm: selected = " + selectedArm.name);
+            return selectedArm.name;
+        } else {
+            Log.e(TAG, "getArm: No arm was selected");
+            return "default_arm";
+        }
     }
 
     // Selects an arm from a list of arms