From 321ec548a86e858b71977188d4557ebb15533544 Mon Sep 17 00:00:00 2001
From: Bill Huang <bill.huang2001@gmail.com>
Date: Tue, 6 Aug 2024 15:43:19 +0800
Subject: [PATCH] dev: revert back to the old workstep.step's signature

---
 src/evox/workflows/std_workflow.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/evox/workflows/std_workflow.py b/src/evox/workflows/std_workflow.py
index 848f48f9c..c2dba4d97 100644
--- a/src/evox/workflows/std_workflow.py
+++ b/src/evox/workflows/std_workflow.py
@@ -252,7 +252,7 @@ def _step(self, state):
             for monitor in self.registered_hooks["post_step"]:
                 monitor.post_step(state)
 
-            return train_info, state
+            return state
 
         if self.jit_step:
             # the first argument is self, which should be static