Skip to content

Commit 7afe68e

Browse files
Trinkle23897Trinkle23897
committed
Simplify Jumanji official renderer
1 parent 2ce2940 commit 7afe68e

110 files changed

Lines changed: 908 additions & 915 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎envpool/jumanji/_official_render/chex.py‎

Lines changed: 0 additions & 26 deletions
This file was deleted.

‎envpool/jumanji/_official_render/jax/__init__.py‎

Lines changed: 0 additions & 83 deletions
This file was deleted.

‎envpool/jumanji/_official_render/jax/numpy.py‎

Lines changed: 0 additions & 193 deletions
This file was deleted.

‎envpool/jumanji/_official_render/jumanji/__init__.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
# ruff: noqa
2+
# fmt: off
3+
from __future__ import annotations
14
# Copyright 2026 Garena Online Private Limited
25
#
36
# Licensed under the Apache License, Version 2.0 (the "License");

‎envpool/jumanji/_official_render/jumanji/env.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
# ruff: noqa
2+
# fmt: off
13
# Copyright 2026 Garena Online Private Limited
24
#
35
# Licensed under the Apache License, Version 2.0 (the "License");

‎envpool/jumanji/_official_render/jumanji/environments/__init__.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
# ruff: noqa
2+
# fmt: off
3+
from __future__ import annotations
14
# Copyright 2026 Garena Online Private Limited
25
#
36
# Licensed under the Apache License, Version 2.0 (the "License");

‎envpool/jumanji/_official_render/jumanji/environments/commons/__init__.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
# ruff: noqa
2+
# fmt: off
3+
from __future__ import annotations
14
# Copyright 2026 Garena Online Private Limited
25
#
36
# Licensed under the Apache License, Version 2.0 (the "License");

‎envpool/jumanji/_official_render/jumanji/environments/commons/graph_view_utils.py‎

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
# ruff: noqa
2+
# fmt: off
3+
from __future__ import annotations
14
# Copyright 2022 InstaDeep Ltd. All rights reserved.
25
#
36
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -14,7 +17,6 @@
1417

1518
from typing import List, Tuple
1619

17-
import chex
1820
import numpy as np
1921

2022

@@ -34,7 +36,7 @@ def _compute_repulsive_forces(
3436

3537

3638
def _compute_attractive_forces(
37-
graph: chex.Array,
39+
graph: Any,
3840
attractive_forces: np.ndarray,
3941
pos: np.ndarray,
4042
k: float,
@@ -54,7 +56,7 @@ def _compute_attractive_forces(
5456

5557

5658
def spring_layout(
57-
graph: chex.Array, num_nodes: int, seed: int = 42, iterations: int = 100
59+
graph: Any, num_nodes: int, seed: int = 42, iterations: int = 100
5860
) -> List[Tuple[float, float]]:
5961
"""
6062
Compute a 2D spring layout for the given graph using

‎envpool/jumanji/_official_render/jumanji/environments/commons/maze_utils/__init__.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
# ruff: noqa
2+
# fmt: off
3+
from __future__ import annotations
14
# Copyright 2026 Garena Online Private Limited
25
#
36
# Licensed under the Apache License, Version 2.0 (the "License");

‎envpool/jumanji/_official_render/jumanji/environments/commons/maze_utils/maze_generation.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
# ruff: noqa
2+
# fmt: off
3+
from __future__ import annotations
14
# Copyright 2026 Garena Online Private Limited
25
#
36
# Licensed under the Apache License, Version 2.0 (the "License");

0 commit comments

Comments
 (0)