Bugfix in atari_il.py example (wrong type check) [ci skip]

examples/offline/atari_il.py

@@ -77,7 +77,7 @@ def test_il(args: argparse.Namespace = get_args()) -> None:
     space_info = SpaceInfo.from_env(env)
     args.state_shape = space_info.observation_info.obs_shape
     args.action_shape = space_info.action_info.action_shape
-    assert isinstance(args.state_shape, list[int] | tuple[int])
+    assert isinstance(args.state_shape, tuple | list)
     assert len(args.state_shape) == 3
     c, h, w = args.state_shape
     # should be N_FRAMES x H x W