From 14b3f47086a1041995f8975a8c4fc4d8706ac8f1 Mon Sep 17 00:00:00 2001
From: ArthurOtte <82815412+ArthurOtte@users.noreply.github.com>
Date: Wed, 4 Dec 2024 13:09:47 +0100
Subject: [PATCH] Update neuralnet.py

Running mooc_ecmwf_bocquet_brajard_part2.ipynb, I received the error "Exception encountered when calling Periodic1DPadding.call(). name 'np' is not defined". Hopefully fixed this by importing numpy in this file.
---
 tier_3/data_assimilation/model/neuralnet.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tier_3/data_assimilation/model/neuralnet.py b/tier_3/data_assimilation/model/neuralnet.py
index 3578ac8..79569ae 100644
--- a/tier_3/data_assimilation/model/neuralnet.py
+++ b/tier_3/data_assimilation/model/neuralnet.py
@@ -3,6 +3,7 @@
 import pickle
 from tensorflow.keras.models import load_model
 from .lorenz96 import l96
+import numpy as np
 
 class Periodic1DPadding(Layer):
     """Add a periodic padding to the output of a layer
@@ -53,4 +54,4 @@ def __call__(self, q):
         dq_phi = super().__call__(q)
         #dq_nn =  self.scaler_y.inverse_transform(self.nn.predict(self.scaler_x.transform(q), verbose=0).squeeze())
         dq_nn =  self.scaler_y.inverse_transform(self.nn(self.scaler_x.transform(q)).numpy().squeeze())
-        return dq_phi + dq_nn
\ No newline at end of file
+        return dq_phi + dq_nn