From fc4094735cd9f72b83c02147ac186a5d3b4923b0 Mon Sep 17 00:00:00 2001
From: Lewis Dicks <141623206+lewisd-benefex@users.noreply.github.com>
Date: Wed, 27 Sep 2023 13:38:25 +0100
Subject: [PATCH] Fixed model_audit.sql to allow ARIMA_PLUS model types

ARIMA is set to be deprecated so there should be an allowance to use both in the interim
---
 macros/hooks/model_audit.sql | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/macros/hooks/model_audit.sql b/macros/hooks/model_audit.sql
index d35fe2e..7d8779c 100644
--- a/macros/hooks/model_audit.sql
+++ b/macros/hooks/model_audit.sql
@@ -59,6 +59,16 @@ arima:
         - duration_ms
         - array(select as struct null as centroid_id, cast(null as float64) as cluster_radius, null as cluster_size)
     feature_info: *default_feature_info
+arima_plus:
+    training_info:
+        - training_run
+        - iteration
+        - cast(null as float64) as loss
+        - cast(null as float64) as eval_loss
+        - cast(null as float64) as learning_rate
+        - duration_ms
+        - array(select as struct null as centroid_id, cast(null as float64) as cluster_radius, null as cluster_size)
+    feature_info: *default_feature_info
 linear_reg:
     training_info: *default_training_info
     feature_info: *default_feature_info