mettagrid 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of mettagrid might be problematic. Click here for more details.
- mettagrid/config/room/ascii.py +1 -0
- mettagrid/mettagrid_env.py +2 -7
- {mettagrid-0.1.3.dist-info → mettagrid-0.1.4.dist-info}/METADATA +1 -1
- {mettagrid-0.1.3.dist-info → mettagrid-0.1.4.dist-info}/RECORD +6 -7
- mettagrid/config/config.py +0 -47
- {mettagrid-0.1.3.dist-info → mettagrid-0.1.4.dist-info}/LICENSE +0 -0
- {mettagrid-0.1.3.dist-info → mettagrid-0.1.4.dist-info}/WHEEL +0 -0
mettagrid/config/room/ascii.py
CHANGED
mettagrid/mettagrid_env.py
CHANGED
|
@@ -5,14 +5,11 @@ from typing import Any, Dict
|
|
|
5
5
|
import gymnasium as gym
|
|
6
6
|
import hydra
|
|
7
7
|
import numpy as np
|
|
8
|
-
from torch import sub
|
|
9
|
-
from mettagrid.config.config import make_odd
|
|
10
8
|
import pufferlib
|
|
11
9
|
from omegaconf import OmegaConf, DictConfig
|
|
12
10
|
|
|
13
11
|
from mettagrid.mettagrid_c import MettaGrid # pylint: disable=E0611
|
|
14
|
-
|
|
15
|
-
from util.config import config_from_path
|
|
12
|
+
|
|
16
13
|
class MettaGridEnv(pufferlib.PufferEnv, gym.Env):
|
|
17
14
|
def __init__(self, env_cfg: DictConfig, render_mode: str, buf=None, **kwargs):
|
|
18
15
|
self._render_mode = render_mode
|
|
@@ -167,13 +164,11 @@ class MettaGridEnv(pufferlib.PufferEnv, gym.Env):
|
|
|
167
164
|
def close(self):
|
|
168
165
|
pass
|
|
169
166
|
|
|
170
|
-
|
|
171
167
|
def make_env_from_cfg(cfg_path: str, *args, **kwargs):
|
|
172
|
-
cfg =
|
|
168
|
+
cfg = OmegaConf.load(cfg_path)
|
|
173
169
|
env = MettaGridEnv(cfg, *args, **kwargs)
|
|
174
170
|
return env
|
|
175
171
|
|
|
176
|
-
|
|
177
172
|
def oc_uniform(min_val, max_val, center, *, _root_):
|
|
178
173
|
sampling = _root_.get("sampling", 0)
|
|
179
174
|
if sampling == 0:
|
|
@@ -185,8 +185,7 @@ mettagrid/actions/swap.pxd,sha256=mS-fT6rY9ug0c5lXkWeaLOkX9la0ZHSr3ffGz-aZBDY,10
|
|
|
185
185
|
mettagrid/actions/swap.pyx,sha256=SXgn5mCUFBt4P3BOe61bSBn4zdBRHxBGc6ePekpWQnM,1429
|
|
186
186
|
mettagrid/actions/use.cpython-311-darwin.so,sha256=jrGpfVPqU-LROL6a-6-C7_94B_ADYCcrgEneXFpbZ10,186760
|
|
187
187
|
mettagrid/base_encoder.cpython-311-darwin.so,sha256=5YeHMdKCIGaxm71xEf_zB5shdgMQs_dC45nzJ399LjY,188064
|
|
188
|
-
mettagrid/config/
|
|
189
|
-
mettagrid/config/room/ascii.py,sha256=FxcA8eGTc-haNLDeT5fePi7QFtBYGxB-sIXJLbM2dik,793
|
|
188
|
+
mettagrid/config/room/ascii.py,sha256=bahuAvBmP71C_ypMdmGgeORGyxNtOHACWyUVROoUask,810
|
|
190
189
|
mettagrid/config/room/cognitive_evals/barriermaze.py,sha256=MPjGjY-SU06D57KNR-g9cWRIreXYReppp7G3cVxGcH8,4647
|
|
191
190
|
mettagrid/config/room/cognitive_evals/corridor_maze.py,sha256=3KHz16kH6rYE-z0pKlqL1L4rMailcU50zTMeJze9s6U,3539
|
|
192
191
|
mettagrid/config/room/cognitive_evals/cylinder.py,sha256=qFYwrS-_N1njEe0wajcM563yCIrLuBSmrVMYIWNYqRc,4027
|
|
@@ -220,7 +219,7 @@ mettagrid/grid_object.pxd,sha256=Plp85O4e0HsZUt5Bl4vrKRIE1ac3E1rwysj2WRgYaUM,882
|
|
|
220
219
|
mettagrid/grid_object.pyx,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
221
220
|
mettagrid/mettagrid.pyx,sha256=Q60WkCskp_pfmux2oVJqEmULSdmqnC49rc8peg7t-CY,9457
|
|
222
221
|
mettagrid/mettagrid_c.cpython-311-darwin.so,sha256=wxA1YIcF9XZ-gyatD1y-5ZkLCJZsKL4ZNrBi9emmyyM,334000
|
|
223
|
-
mettagrid/mettagrid_env.py,sha256=
|
|
222
|
+
mettagrid/mettagrid_env.py,sha256=HLEcfNUb97vGi_wPn1W0sjK34mx04jCvVIoCzFRNIKU,6331
|
|
224
223
|
mettagrid/objects/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
225
224
|
mettagrid/objects/agent.cpython-311-darwin.so,sha256=BAd--z_jI_Np5sOgV9TV9FjgXW0KB4_3qcpQG9sjNRM,57608
|
|
226
225
|
mettagrid/objects/agent.hpp,sha256=JCbToyoajLPtMpobUnakwQMtY-hg_WJsaKAKtsawL-c,4279
|
|
@@ -320,7 +319,7 @@ tests/regen_track_last_action_trace.py,sha256=a1QdBRkvvBC7C0ZANQiwknb3QFQa6iHkNm
|
|
|
320
319
|
tests/test_basic.py,sha256=moICNw_9L_lfSQRT2k7LWdEIEC9x2kwK1o46ClVvz2Q,4861
|
|
321
320
|
tests/test_kinship.py,sha256=TvyPBbVazgYMGP-KbkWqKoo1QO1d3iT2uPwccyKypss,2406
|
|
322
321
|
tests/test_leaks.py,sha256=Uo-L5h68knTDVcnSUZ8rNXeb-rcMTOqZNOB0SYTC5_I,570
|
|
323
|
-
mettagrid-0.1.
|
|
324
|
-
mettagrid-0.1.
|
|
325
|
-
mettagrid-0.1.
|
|
326
|
-
mettagrid-0.1.
|
|
322
|
+
mettagrid-0.1.4.dist-info/LICENSE,sha256=_0ZygdDOImEUQjXap6LgAetwhuvlOfOqcMDmNlAdjLo,1070
|
|
323
|
+
mettagrid-0.1.4.dist-info/METADATA,sha256=nkDgfNsFwyu7hhzg_OtLV4dkjhxl9-zZJc6TUYWUkJU,7166
|
|
324
|
+
mettagrid-0.1.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
|
325
|
+
mettagrid-0.1.4.dist-info/RECORD,,
|
mettagrid/config/config.py
DELETED
|
@@ -1,47 +0,0 @@
|
|
|
1
|
-
import os
|
|
2
|
-
import random
|
|
3
|
-
import signal
|
|
4
|
-
|
|
5
|
-
import numpy as np
|
|
6
|
-
import torch
|
|
7
|
-
from omegaconf import OmegaConf
|
|
8
|
-
from rich import traceback
|
|
9
|
-
import warnings
|
|
10
|
-
|
|
11
|
-
warnings.warn("This config.py file is deprecated", DeprecationWarning)
|
|
12
|
-
|
|
13
|
-
def uniform(min_val, max_val, center, *, _root_):
|
|
14
|
-
sampling = _root_["sampling"]
|
|
15
|
-
if sampling == 0:
|
|
16
|
-
return center
|
|
17
|
-
|
|
18
|
-
center = (max_val + min_val) // 2
|
|
19
|
-
# Calculate the available range on both sides of the center
|
|
20
|
-
left_range = center - min_val
|
|
21
|
-
right_range = max_val - center
|
|
22
|
-
|
|
23
|
-
# Scale the ranges based on the sampling parameter
|
|
24
|
-
scaled_left = min(left_range, sampling * left_range)
|
|
25
|
-
scaled_right = min(right_range, sampling * right_range)
|
|
26
|
-
|
|
27
|
-
# Generate a random value within the scaled range
|
|
28
|
-
val = np.random.uniform(center - scaled_left, center + scaled_right)
|
|
29
|
-
|
|
30
|
-
# Clip to ensure we stay within [min_val, max_val]
|
|
31
|
-
val = np.clip(val, min_val, max_val)
|
|
32
|
-
|
|
33
|
-
# Return integer if the original values were integers
|
|
34
|
-
return int(round(val)) if isinstance(center, int) else val
|
|
35
|
-
|
|
36
|
-
def choose(*args):
|
|
37
|
-
return random.choice(args)
|
|
38
|
-
|
|
39
|
-
def div(a, b):
|
|
40
|
-
return a // b
|
|
41
|
-
|
|
42
|
-
def sub(a, b):
|
|
43
|
-
return a - b
|
|
44
|
-
|
|
45
|
-
def make_odd(a):
|
|
46
|
-
return max(3, a // 2 * 2 + 1)
|
|
47
|
-
|
|
File without changes
|
|
File without changes
|