From 76a8f8396bd8e4620e47cac86efe9894201ef83c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Sat, 19 Oct 2024 23:38:42 -0500
Subject: [PATCH] fix: script types

---
 packages/scripts/src/averages.ts | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/packages/scripts/src/averages.ts b/packages/scripts/src/averages.ts
index 7d54ab84a..b6d869a6e 100644
--- a/packages/scripts/src/averages.ts
+++ b/packages/scripts/src/averages.ts
@@ -1,7 +1,6 @@
 import {
   AverageDefinitions,
   AverageDefinitionsEntry,
-  AverageDefinitionsEntrySubPartial,
   AverageDefinitionsManifest,
   IndividualTankStats,
   REGIONS,
@@ -9,7 +8,7 @@ import {
   Samples,
   averageDefinitionsAllStatsKeys,
   emptySamples,
-  encodeProtobufToBase64,
+  encodePB64,
   getAccountInfo,
   getTankStats,
   idToRegion,
@@ -24,6 +23,17 @@ interface DataPoint {
   w: number;
 }
 
+type OptionalSecondLevel<T> = {
+  [P in keyof T]: P extends 'samples'
+    ? T[P]
+    : {
+        [K in keyof T[P]]?: T[P][K];
+      };
+};
+
+type AverageDefinitionsEntrySubPartial =
+  OptionalSecondLevel<AverageDefinitionsEntry>;
+
 const MINUTE = 60 * 1000;
 const HOUR = 60 * MINUTE;
 const DAY = 24 * HOUR;
@@ -225,11 +235,7 @@ async function postWork() {
   commitAssets('averages', [
     {
       path: `averages/${latest}.pb`,
-      content: await encodeProtobufToBase64(
-        'average_definitions',
-        'blitzkit.AverageDefinitions',
-        averageDefinitions,
-      ),
+      content: encodePB64(AverageDefinitions, averageDefinitions),
       encoding: 'base64',
     },
     {