Update inference.py

inference.py

@@ -31,9 +31,9 @@ with open(params_path, 'r') as f:
 
 device = params['device']
 
-# --- تحميل النموذج ---
+# --- تحميل النموذج بعد ضبط input_size = 4 ---
 model = Model(
-    input_size = 5,
+    input_size = 4,
     embedding_size=params["embedding_size"],
     decoder_input_size=params["decoder_input_size"]
 )
@@ -52,8 +52,10 @@ delivery_lat = df['delivery_gps_lat'].to_numpy()
 coords = np.stack([pickup_lng, pickup_lat, delivery_lng, delivery_lat], axis=1)
 coords_tensor = torch.tensor(coords, dtype=torch.float32).unsqueeze(0).to(device)
 
-# --- تجهيز Batch ---
-batch = [{"coords": coords_tensor}]
+# --- تجهيز Batch كـ Tuple لتجنب unpacking error في Actor ---
+graph_data = {"coords": coords_tensor}
+fleet_data = {"dummy": torch.tensor([0])}  # يمكن تعديل هذا لاحقاً عند الحاجة
+batch = (graph_data, fleet_data)
 
 # --- تهيئة الممثل والـ NN ---
 actor = Actor(