check for subway

mbta · Jul 10, 2024 · 6a2f5c0 · 6a2f5c0
1 parent 78dff58
commit 6a2f5c0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/alerts/alert.ex b/lib/alerts/alert.ex
@@ -149,10 +149,10 @@ defmodule Alerts.Alert do
 
   defp set_direction_ids(%__MODULE__{} = alert), do: alert
 
-  defp set_direction_id(%InformedEntity{} = entity) do
+  defp set_direction_id(%InformedEntity{direction_id: nil, route_type: 1} = entity) do
     stop = Stops.Repo.get(entity.stop)
 
-    if entity.direction_id == nil && stop.child_ids == [] do
+    if stop.child_ids == [] do
       direction_id =
         stop.id
         |> RoutePatterns.Repo.by_stop_id()
@@ -166,6 +166,8 @@ defmodule Alerts.Alert do
     end
   end
 
+  defp set_direction_id(entity), do: entity
+
   @spec set_priority(map) :: map
   defp set_priority(%__MODULE__{} = alert) do
     %__MODULE__{alert | priority: Priority.priority(alert)}