From 63cd66d53cfa5e641c57530b117490ea7cd6b13f Mon Sep 17 00:00:00 2001
From: Florian Pinault <Florian.Pinault@ecmwf.int>
Date: Tue, 8 Oct 2024 15:12:06 +0000
Subject: [PATCH] example with a model

---
 src/anemoi/models/preprocessing/normalizer.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/anemoi/models/preprocessing/normalizer.py b/src/anemoi/models/preprocessing/normalizer.py
index d3a816c..2878c85 100644
--- a/src/anemoi/models/preprocessing/normalizer.py
+++ b/src/anemoi/models/preprocessing/normalizer.py
@@ -11,7 +11,7 @@
 import warnings
 from typing import Optional
 
-from anemoi.datasets import AnemoiState
+from anemoi.utils.data_structures import NumpyNestedAnemoiTensor
 
 import numpy as np
 import torch
@@ -53,10 +53,10 @@ def __init__(
         mean = statistics["mean"]
         stdev = statistics["stdev"]
         
-        mean = AnemoiState(mean)
-        stdev = AnemoiState(stdev)
-        minimum = AnemoiState(minimum)
-        maximum = AnemoiState(maximum)
+        mean = NumpyNestedAnemoiTensor(mean)
+        stdev = NumpyNestedAnemoiTensor(stdev)
+        minimum = NumpyNestedAnemoiTensor(minimum)
+        maximum = NumpyNestedAnemoiTensor(maximum)
 
         self._validate_normalization_inputs(name_to_index_training_input, minimum, maximum, mean, stdev)