bug fix

visualiser/hkust_bnb_visualiser.py

@@ -30,8 +30,7 @@ from constant.hkust_bnb_constant import (
     TRAFFIC_SPOT_INFO_TEMPLATE,
     RELEVANCE_INFO_TEMPLATE,
     POPUP_CONTENT_TEMPLATE,
-    MAP_SCRIPT,
-    NO_DISCOUNT_INFO_TEMPLATE
+    MAP_SCRIPT
 )
 
 class HKUSTBNBVisualiser:
@@ -389,16 +388,10 @@ class HKUSTBNBVisualiser:
             discount_info = ""
             discounted_price = row['price']
 
-            # Inside the create_map_and_data method
             if row['id'] in all_nearest_traffic_spots:
                 nearest_spot, distance = all_nearest_traffic_spots[row['id']]
                 discount_rate = nearest_spot.get_discount_rate()
 
-                # Calculate average vehicle count safely
-                avg_vehicle_count = nearest_spot.avg_vehicle_count
-                if avg_vehicle_count is None:
-                    avg_vehicle_count = "N/A"  # Or calculate it directly here if needed
-
                 if discount_rate > 0:
                     discounted_price = row['price'] * (1 - discount_rate)
                     discount_percentage = int(discount_rate * 100)
@@ -407,12 +400,7 @@ class HKUSTBNBVisualiser:
                         discount_percentage=discount_percentage,
                         original_price=row['price'],
                         discounted_price=discounted_price,
-                        avg_vehicle_count=avg_vehicle_count,
-                        observation_count=len(nearest_spot.dataset_rows)
-                    )
-                else:
-                    discount_info = NO_DISCOUNT_INFO_TEMPLATE.format(
-                        avg_vehicle_count=avg_vehicle_count,
+                        avg_vehicle_count=nearest_spot.avg_vehicle_count,
                         observation_count=len(nearest_spot.dataset_rows)
                     )