From 87b1be75ddaadb32961026a00de24068a9a69a01 Mon Sep 17 00:00:00 2001 From: Gaspard Lambrechts <57069365+glambrechts@users.noreply.github.com> Date: Tue, 20 Jan 2026 15:40:59 -0500 Subject: [PATCH 1/2] Ensure 4-dimensional image is loaded --- craftax/craftax/constants.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/craftax/craftax/constants.py b/craftax/craftax/constants.py index 41bcc175..404ec259 100644 --- a/craftax/craftax/constants.py +++ b/craftax/craftax/constants.py @@ -593,7 +593,7 @@ def achievement_mapping(achievement_value): # TEXTURES def load_texture(filename, block_pixel_size): filename = os.path.join(pathlib.Path(__file__).parent.resolve(), "assets", filename) - img = iio.imread(filename) + img = iio.imread(filename, mode="RGBA") jnp_img = jnp.array(img).astype(int) assert jnp_img.shape[:2] == (16, 16) From 3d333be508c9df0f4c1551a8617c932ad39b8e9c Mon Sep 17 00:00:00 2001 From: Gaspard Lambrechts <57069365+glambrechts@users.noreply.github.com> Date: Wed, 21 Jan 2026 11:32:15 -0500 Subject: [PATCH 2/2] Ensure 4-dimensional images are loaded and usable --- craftax/craftax_classic/constants.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/craftax/craftax_classic/constants.py b/craftax/craftax_classic/constants.py index 1a495d6d..44f725ac 100644 --- a/craftax/craftax_classic/constants.py +++ b/craftax/craftax_classic/constants.py @@ -132,7 +132,7 @@ class Achievement(Enum): # TEXTURES def load_texture(filename, block_pixel_size, clamp_alpha=True): filename = os.path.join(pathlib.Path(__file__).parent.resolve(), "assets", filename) - img = iio.imread(filename) + img = iio.imread(filename, mode="RGBA") jnp_img = jnp.array(img).astype(int) assert jnp_img.shape[:2] == (16, 16) @@ -176,12 +176,12 @@ def load_all_textures(block_pixel_size): block_textures = jnp.array( [ - load_texture("debug_tile.png", block_pixel_size), + load_texture("debug_tile.png", block_pixel_size)[:, :, :3], jnp.ones((block_pixel_size, block_pixel_size, 3), dtype=jnp.int32) * 128, - load_texture("grass.png", block_pixel_size), - load_texture("water.png", block_pixel_size), - load_texture("stone.png", block_pixel_size), - load_texture("tree.png", block_pixel_size), + load_texture("grass.png", block_pixel_size)[:, :, :3], + load_texture("water.png", block_pixel_size)[:, :, :3], + load_texture("stone.png", block_pixel_size)[:, :, :3], + load_texture("tree.png", block_pixel_size)[:, :, :3], load_texture("wood.png", block_pixel_size)[:, :, :3], load_texture("path.png", block_pixel_size)[:, :, :3], load_texture("coal.png", block_pixel_size)[:, :, :3],