WIP: update to include sectors!

[browser-rts.git] / src / render / unit-training.ts

diff --git a/src/render/unit-training.ts b/src/render/unit-training.ts
index dfbbb1bc40fee300ca5c66558f9e9059c8e65db0..5b4ffcf026cd3d0613af6f76c5ac3da2f73c0c9d 100644 (file)
--- a/src/render/unit-training.ts
+++ b/src/render/unit-training.ts
@@ -1,5 +1,5 @@
 import _ from "lodash";
-import { City } from "../repository/city";
+import { CityWithLocation } from "../repository/city";
 import { UnitTrainingQueue } from "../repository/training-queue";
 import { Unit } from "../repository/unit";
 
@@ -13,7 +13,7 @@ function progressBar(current, max): string {
     `;
 }
 
-export function renderUnitTraining(city: City, units: Unit[], trainingQueues: UnitTrainingQueue[]): string {
+export function renderUnitTraining(city: CityWithLocation, units: Unit[], trainingQueues: UnitTrainingQueue[]): string {
     const unit = _.keyBy(units, 'slug');
     let html = `
     <div hx-trigger="reload-unit-training, every 600s" hx-get="/poll/unit-training">