From 1cd26427a3bf3a9e1425398211833fe7144fe003 Mon Sep 17 00:00:00 2001
From: Baudouin Raoult <baudouin.raoult@ecmwf.int>
Date: Wed, 23 Oct 2024 07:11:34 +0000
Subject: [PATCH] update

---
 dev/dev.py                           | 2 +-
 src/anemoi/transform/sources/mars.py | 7 ++++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/dev/dev.py b/dev/dev.py
index 5d51de1..859be27 100644
--- a/dev/dev.py
+++ b/dev/dev.py
@@ -50,4 +50,4 @@
     print(f)
 
 
-ipipe = pipeline.to_infernece()
+# ipipe = pipeline.to_infernece()
diff --git a/src/anemoi/transform/sources/mars.py b/src/anemoi/transform/sources/mars.py
index 5f627fb..45b8ee7 100644
--- a/src/anemoi/transform/sources/mars.py
+++ b/src/anemoi/transform/sources/mars.py
@@ -25,10 +25,15 @@ def forward(self, data):
 
     def __ror__(self, data):
 
-        class Input:
+        this = self
+
+        class Input(Source):
             def __init__(self, data):
                 self.data = data
 
+            def forward(self, data):
+                return this.forward(self.data)
+
         return Input(data)