From 81e7ff8d095e8b740f157f4247047fcfd0b4473f Mon Sep 17 00:00:00 2001 From: Sven Mika Date: Tue, 11 Jun 2024 11:50:05 +0200 Subject: [PATCH] [RLlib] Fix CI test: `test_io.py`. (#45839) --- rllib/tests/test_io.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rllib/tests/test_io.py b/rllib/tests/test_io.py index 60f68fba96333..027e1382cc26a 100644 --- 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()