Beds fix

mappingservice/routers/room.py

@@ -139,16 +139,17 @@ async def predict_all(request: Request, room_data: RoomData):
     feature_pred_results = feature_future.result()["feature_prediction"]
     view_pred_results = view_future.result()["view_prediction"]
 
-    bed_predictions = [
-        bed_data for bed_data in room_data.beds if bed_data and bed_data.type is not None and bed_data.count is not None  # noqa: E501
-    ]
+    bed_predictions = room_data.beds
 
-    if not bed_predictions:
+    if not room_data.beds:
         logger.debug("No bed data provided or valid; extracting from description.")
         extracted_beds = get_bed_predictions(room_data.room_description)
         if extracted_beds:
             bed_predictions.extend(extracted_beds)
 
+    if not isinstance(bed_predictions, list):
+        bed_predictions = [bed_predictions]
+
     end_time = time.time()
     total_time = end_time - start_time
     logger.info(f"Total processing time: {total_time:.3f} seconds")