fix

Author: HydrogenSulfate

examples/cylinder/3d_unsteady_discrete/README.md

@@ -255,7 +255,7 @@ def function_with_eval_state(self, *args, **kwargs):
             self.model.eval()
 
         # run func in eval mode
-        func()
+        func(self, *args, **kwargs)
 
         # restore state
         if train_state: