[RLlib] Fix CI test: test_io.py. (#45839)

ray-project · Jun 11, 2024 · 81e7ff8 · 81e7ff8
1 parent 94937a1
commit 81e7ff8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rllib/tests/test_io.py b/rllib/tests/test_io.py
@@ -263,9 +263,9 @@ def input_creator(ioctx: IOContext) -> InputReader:
                 config.offline_data(input_config={"input_files": self.test_dir + fw})
                 algo = config.build()
                 result = algo.train()
-                self.assertEqual(result[f"{NUM_ENV_STEPS_SAMPLED_LIFETIME}"], 4000)
+                self.assertEqual(result[NUM_ENV_STEPS_SAMPLED_LIFETIME], 4000)
                 self.assertTrue(
-                    np.isnan(result[f"{ENV_RUNNER_RESULTS}/{EPISODE_RETURN_MEAN}"])
+                    np.isnan(result[ENV_RUNNER_RESULTS][EPISODE_RETURN_MEAN])
                 )
                 algo.stop()