Updated the finder_api to use the predictions repo pattern

mbta · May 8, 2024 · 6bfb63a · 6bfb63a
1 parent f527f74
commit 6bfb63a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/dotcom_web/controllers/schedule/finder_api.ex b/lib/dotcom_web/controllers/schedule/finder_api.ex
@@ -20,6 +20,7 @@ defmodule DotcomWeb.ScheduleController.FinderApi do
   require Logger
 
   @route_patterns_repo Application.compile_env!(:dotcom, :repo_modules)[:route_patterns]
+  @predictions_repo Application.compile_env!(:dotcom, :repo_modules)[:predictions]
 
   @type react_keys :: :date | :direction | :is_current
   @type react_strings :: [{react_keys, String.t()}]
@@ -217,7 +218,8 @@ defmodule DotcomWeb.ScheduleController.FinderApi do
       direction_id: direction_id
     ]
 
-    predictions_fn = Map.get(conn.assigns, :predictions_fn, &Predictions.Repo.all/1)
+    predictions_fn =
+      Map.get(conn.assigns, :predictions_fn, Function.capture(@predictions_repo, :all, 1))
 
     predictions =
       if current_service?,