From 86ff03a2086c871d3dae1d5c050e247418b25964 Mon Sep 17 00:00:00 2001
From: HomerW <hrwalke@gmail.com>
Date: Mon, 20 May 2024 09:26:16 -0700
Subject: [PATCH] fix order of supply rng and partial in example

---
 examples/04_eval_finetuned_on_robot.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/04_eval_finetuned_on_robot.py b/examples/04_eval_finetuned_on_robot.py
index ddd8e5d9..32cca713 100644
--- a/examples/04_eval_finetuned_on_robot.py
+++ b/examples/04_eval_finetuned_on_robot.py
@@ -129,8 +129,8 @@ def sample_actions(
         # remove batch dim
         return actions[0]
 
-    policy_fn = partial(
-        supply_rng(
+    policy_fn = supply_rng(
+        partial(
             sample_actions,
             model,
             argmax=FLAGS.deterministic,