mettagrid 0.2__tar.gz → 0.2.0.1__tar.gz

{mettagrid-0.2 → mettagrid-0.2.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mettagrid
-Version: 0.2
+Version: 0.2.0.1
 Summary: A fast grid-based open-ended MARL environment
 Author-email: David Bloomin <daveey@gmail.com>
 License-Expression: MIT
@@ -18,7 +18,7 @@ Requires-Dist: google-auth-oauthlib>=1.0.0
 Requires-Dist: gymnasium==0.29.1
 Requires-Dist: matplotlib>=3.10.3
 Requires-Dist: numpy<2,>=1.26.4
-Requires-Dist: omegaconf>=2.4.0.dev3
+Requires-Dist: omegaconf>=2.3.0
 Requires-Dist: duckdb>=1.3.0
 Requires-Dist: pettingzoo<1.25,>=1.24.1
 Requires-Dist: pufferlib<3.1.0,>=3.0.0

{mettagrid-0.2 → mettagrid-0.2.0.1}/bazel_build.py

@@ -23,6 +23,7 @@ from setuptools.build_meta import (
     prepare_metadata_for_build_editable,
     prepare_metadata_for_build_wheel,
 )
+from setuptools.dist import Distribution
 
 PROJECT_ROOT = Path(__file__).resolve().parent
 
@@ -31,9 +32,7 @@ def _run_bazel_build() -> None:
     """Run Bazel build to compile the C++ extension."""
     # Check if bazel is available
     if shutil.which("bazel") is None:
-        raise RuntimeError(
-            "Bazel is required to build metta-mettagrid. Please install Bazel: https://bazel.build/install"
-        )
+        raise RuntimeError("Bazel is required to build mettagrid. Please install Bazel: https://bazel.build/install")
 
     # Determine build configuration from environment
     debug = os.environ.get("DEBUG", "").lower() in ("1", "true", "yes")
@@ -52,6 +51,7 @@ def _run_bazel_build() -> None:
         "bazel",
         "build",
         f"--config={config}",
+        "--verbose_failures",
         "//:mettagrid_c",
     ]
 
@@ -59,7 +59,10 @@ def _run_bazel_build() -> None:
     result = subprocess.run(cmd, cwd=PROJECT_ROOT, capture_output=True, text=True)
 
     if result.returncode != 0:
-        print(f"Bazel build failed:\n{result.stderr}", file=sys.stderr)
+        print("Bazel build failed. STDERR:", file=sys.stderr)
+        print(result.stderr, file=sys.stderr)
+        print("Bazel build STDOUT:", file=sys.stderr)
+        print(result.stdout, file=sys.stderr)
         raise RuntimeError("Bazel build failed")
 
     # Copy the built extension to the package directory
@@ -92,7 +95,14 @@ def _run_bazel_build() -> None:
 def build_wheel(wheel_directory, config_settings=None, metadata_directory=None):
     """Build a wheel, compiling the C++ extension with Bazel first."""
     _run_bazel_build()
-    return _build_wheel(wheel_directory, config_settings, metadata_directory)
+    # Ensure wheel is tagged as non-pure (platform-specific) since we bundle a native extension
+    # Setuptools/wheel derive purity from Distribution.has_ext_modules(). Monkeypatch to force True.
+    original_has_ext_modules = Distribution.has_ext_modules
+    try:
+        Distribution.has_ext_modules = lambda self: True  # type: ignore[assignment]
+        return _build_wheel(wheel_directory, config_settings, metadata_directory)
+    finally:
+        Distribution.has_ext_modules = original_has_ext_modules
 
 
 def build_editable(wheel_directory, config_settings=None, metadata_directory=None):

{mettagrid-0.2 → mettagrid-0.2.0.1}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "bazel_build"
 backend-path = ["."]
 
 [project]
-name = "mettagrid"
-version = "0.2"
+name = "mettagrid" # Do not change! Has to be 'mettagrid' for our PyPi package
+version = "0.2.0.1"
 description = "A fast grid-based open-ended MARL environment"
 authors = [{ name = "David Bloomin", email = "daveey@gmail.com" }]
 requires-python = "==3.11.7"
@@ -29,7 +29,7 @@ dependencies = [
   "gymnasium==0.29.1",
   "matplotlib>=3.10.3",
   "numpy>=1.26.4,<2",
-  "omegaconf>=2.4.0.dev3",
+  "omegaconf>=2.3.0",
   "duckdb>=1.3.0",
   "pettingzoo>=1.24.1,<1.25",
   "pufferlib>=3.0.0,<3.1.0",

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/metta/mettagrid/actions/get_output.hpp

@@ -26,11 +26,7 @@ protected:
     // Once we generalize this to `get`, we should be able to get from any HasInventory object, which
     // should include agents. That's (e.g.) why we're checking inventory_is_accessible.
     Converter* converter = dynamic_cast<Converter*>(_grid->object_at(target_loc));
-    Box* box = dynamic_cast<Box*>(_grid->object_at(target_loc));
-    if (!converter && !box) {
-      return false;
-    }
-    // If converter, get output from converter
+
     if (converter) {
       if (!converter->inventory_is_accessible()) {
         return false;
@@ -55,32 +51,6 @@ protected:
       }
       return resources_taken;
     }
-
-    // If box, pick up box if allowed
-    if (box) {
-      if (actor->agent_id == box->creator_agent_id) {
-        // Creator cannot open their own box
-        return false;
-      }
-      // If the creator of the box is an agent, return blue battery to creator and penalize creator
-      if (box->creator_agent_id != 255) {
-        Agent* creator = dynamic_cast<Agent*>(_grid->object(box->creator_agent_object_id));
-        if (!creator) {
-          return false;
-        }
-        // Return required resources to create box to creator inventory
-        for (const auto& [item, amount] : box->returned_resources) {
-          if (amount > 0) {
-            creator->update_inventory(item, amount);
-          }
-        }
-      }
-
-      // Reward the agent for opening the box and teleport back to top-left corner
-      _grid->ghost_move_object(box->id, GridLocation(0, 0, GridLayer::ObjectLayer));
-      actor->stats.add("box.opened", 1.0f);
-      return true;
-    }
     return false;
   }
 };

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/metta/mettagrid/builder/envs.py

@@ -112,7 +112,7 @@ def make_navigation(num_agents: int) -> MettaGridConfig:
             resource_names=["heart"],
             actions=ActionsConfig(
                 move=ActionConfig(),
-                rotate=ActionConfig(),
+                rotate=ActionConfig(enabled=False),
                 get_items=ActionConfig(),
             ),
             agent=AgentConfig(
@@ -149,7 +149,7 @@ def make_navigation_sequence(num_agents: int) -> MettaGridConfig:
             resource_names=["heart", "ore_red", "battery_red"],
             actions=ActionsConfig(
                 move=ActionConfig(),
-                rotate=ActionConfig(),
+                rotate=ActionConfig(enabled=False),
                 get_items=ActionConfig(),
             ),
             agent=AgentConfig(
@@ -173,7 +173,7 @@ def make_navigation_sequence(num_agents: int) -> MettaGridConfig:
 
 
 def make_icl_resource_chain(
-    num_agents: int, max_steps, game_objects: dict, map_builder_objects: dict
+    num_agents: int, max_steps, game_objects: dict, map_builder_objects: dict, width: int = 6, height: int = 6
 ) -> MettaGridConfig:
     game_objects["wall"] = empty_converters.wall
     cfg = MettaGridConfig(
@@ -184,7 +184,7 @@ def make_icl_resource_chain(
             map_builder=MapGen.Config(
                 instances=num_agents,
                 instance_map=PerimeterInContextMapBuilder.Config(
-                    agents=1, width=6, height=6, objects=map_builder_objects
+                    agents=1, width=width, height=height, objects=map_builder_objects
                 ),
             ),
             actions=ActionsConfig(

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/metta/mettagrid/mettagrid_c.cpp

@@ -16,7 +16,6 @@
 #include "actions/get_output.hpp"
 #include "actions/move.hpp"
 #include "actions/noop.hpp"
-#include "actions/place_box.hpp"
 #include "actions/put_recipe_items.hpp"
 #include "actions/rotate.hpp"
 #include "actions/swap.hpp"
@@ -24,7 +23,6 @@
 #include "grid.hpp"
 #include "hash.hpp"
 #include "objects/agent.hpp"
-#include "objects/box.hpp"
 #include "objects/constants.hpp"
 #include "objects/converter.hpp"
 #include "objects/converter_config.hpp"
@@ -85,8 +83,6 @@ MettaGrid::MettaGrid(const GameConfig& game_config, const py::list map, unsigned
   for (const auto& [action_name, action_config] : game_config.actions) {
     if (action_name == "put_items") {
       _action_handlers.push_back(std::make_unique<PutRecipeItems>(*action_config));
-    } else if (action_name == "place_box") {
-      _action_handlers.push_back(std::make_unique<PlaceBox>(*action_config));
     } else if (action_name == "get_items") {
       _action_handlers.push_back(std::make_unique<GetOutput>(*action_config));
     } else if (action_name == "noop") {
@@ -168,14 +164,6 @@ MettaGrid::MettaGrid(const GameConfig& game_config, const py::list map, unsigned
         continue;
       }
 
-      const BoxConfig* box_config = dynamic_cast<const BoxConfig*>(object_cfg);
-      if (box_config) {
-        Box* box = new Box(r, c, *box_config, 255, 255);  // Default creator values
-        _grid->add_object(box);
-        _stats->incr("objects." + cell);
-        continue;
-      }
-
       const ConverterConfig* converter_config = dynamic_cast<const ConverterConfig*>(object_cfg);
       if (converter_config) {
         // Create a new ConverterConfig with the recipe offsets from the observation encoder
@@ -204,14 +192,6 @@ MettaGrid::MettaGrid(const GameConfig& game_config, const py::list map, unsigned
         if (_global_obs_config.visitation_counts) {
           agent->init_visitation_grid(height, width);
         }
-        // add agent box
-        if (game_config.objects.contains("box")) {
-          const BoxConfig* local_box_cfg = dynamic_cast<const BoxConfig*>(game_config.objects.at("box").get());
-          if (local_box_cfg) {
-            agent->box = new Box(0, 0, *local_box_cfg, agent->id, static_cast<unsigned char>(agent->agent_id));
-            _grid->ghost_add_object(agent->box);
-          }
-        }
         add_agent(agent);
         _group_sizes[agent->group] += 1;
         continue;
@@ -929,15 +909,6 @@ PYBIND11_MODULE(mettagrid_c, m) {
 
   bind_wall_config(m);
 
-  py::class_<BoxConfig, GridObjectConfig, std::shared_ptr<BoxConfig>>(m, "BoxConfig")
-      .def(py::init<TypeId, const std::string&, const std::map<InventoryItem, InventoryQuantity>&>(),
-           py::arg("type_id"),
-           py::arg("type_name") = "box",
-           py::arg("returned_resources"))
-      .def_readwrite("type_id", &BoxConfig::type_id)
-      .def_readwrite("type_name", &BoxConfig::type_name)
-      .def_readwrite("returned_resources", &BoxConfig::returned_resources);
-
   // ##MettaGridConfig
   // We expose these as much as we can to Python. Defining the initializer (and the object's constructor) means
   // we can create these in Python as AgentConfig(**agent_config_dict). And then we expose the fields individually.

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/metta/mettagrid/mettagrid_c.pyi

@@ -186,6 +186,7 @@ class GameConfig:
         track_movement_metrics: bool = False,
         recipe_details_obs: bool = False,
         allow_diagonals: bool = False,
+        reward_estimates: Optional[dict[str, float]] = None,
     ) -> None: ...
     num_agents: int
     max_steps: int
@@ -200,6 +201,7 @@ class GameConfig:
     track_movement_metrics: bool
     recipe_details_obs: bool
     allow_diagonals: bool
+    reward_estimates: Optional[dict[str, float]]
 
 class MettaGrid:
     obs_width: int

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/metta/mettagrid/mettagrid_c_config.py

@@ -1,13 +1,12 @@
 from metta.mettagrid.mettagrid_c import ActionConfig as CppActionConfig
 from metta.mettagrid.mettagrid_c import AgentConfig as CppAgentConfig
 from metta.mettagrid.mettagrid_c import AttackActionConfig as CppAttackActionConfig
-from metta.mettagrid.mettagrid_c import BoxConfig as CppBoxConfig
 from metta.mettagrid.mettagrid_c import ChangeGlyphActionConfig as CppChangeGlyphActionConfig
 from metta.mettagrid.mettagrid_c import ConverterConfig as CppConverterConfig
 from metta.mettagrid.mettagrid_c import GameConfig as CppGameConfig
 from metta.mettagrid.mettagrid_c import GlobalObsConfig as CppGlobalObsConfig
 from metta.mettagrid.mettagrid_c import WallConfig as CppWallConfig
-from metta.mettagrid.mettagrid_config import AgentConfig, BoxConfig, ConverterConfig, GameConfig, WallConfig
+from metta.mettagrid.mettagrid_config import AgentConfig, ConverterConfig, GameConfig, WallConfig
 
 
 def recursive_update(d, u):
@@ -156,16 +155,6 @@ def convert_to_cpp_game_config(mettagrid_config: dict | GameConfig):
                 swappable=object_config.swappable,
             )
             objects_cpp_params[object_type] = cpp_wall_config
-        elif isinstance(object_config, BoxConfig):
-            returned_resources = game_config.actions.place_box.consumed_resources
-            cpp_box_config = CppBoxConfig(
-                type_id=object_config.type_id,
-                type_name=object_type,
-                returned_resources={
-                    resource_name_to_id[k]: v for k, v in returned_resources.items() if k in resource_name_to_id
-                },
-            )
-            objects_cpp_params[object_type] = cpp_box_config
         else:
             raise ValueError(f"Unknown object type: {object_type}")
 

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/metta/mettagrid/mettagrid_config.hpp

@@ -42,6 +42,7 @@ struct GameConfig {
   bool track_movement_metrics = false;
   bool recipe_details_obs = false;
   bool allow_diagonals = false;
+  std::map<std::string, float> reward_estimates = {};
 };
 
 namespace py = pybind11;
@@ -79,7 +80,8 @@ inline void bind_game_config(py::module& m) {
                     // FEATURE FLAGS
                     bool,
                     bool,
-                    bool>(),
+                    bool,
+                    const std::map<std::string, float>&>(),
            py::arg("num_agents"),
            py::arg("max_steps"),
            py::arg("episode_truncates"),
@@ -95,7 +97,8 @@ inline void bind_game_config(py::module& m) {
            // FEATURE FLAGS
            py::arg("track_movement_metrics"),
            py::arg("recipe_details_obs") = false,
-           py::arg("allow_diagonals") = false)
+           py::arg("allow_diagonals") = false,
+           py::arg("reward_estimates") = std::map<std::string, float>())
       .def_readwrite("num_agents", &GameConfig::num_agents)
       .def_readwrite("max_steps", &GameConfig::max_steps)
       .def_readwrite("episode_truncates", &GameConfig::episode_truncates)
@@ -116,7 +119,8 @@ inline void bind_game_config(py::module& m) {
       // FEATURE FLAGS
       .def_readwrite("track_movement_metrics", &GameConfig::track_movement_metrics)
       .def_readwrite("recipe_details_obs", &GameConfig::recipe_details_obs)
-      .def_readwrite("allow_diagonals", &GameConfig::allow_diagonals);
+      .def_readwrite("allow_diagonals", &GameConfig::allow_diagonals)
+      .def_readwrite("reward_estimates", &GameConfig::reward_estimates);
 }
 
 #endif  // METTAGRID_CONFIG_HPP_

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/metta/mettagrid/mettagrid_config.py

@@ -74,8 +74,7 @@ class ActionsConfig(Config):
     noop: ActionConfig = Field(default_factory=lambda: ActionConfig(enabled=False))
     move: ActionConfig = Field(default_factory=lambda: ActionConfig(enabled=True))  # Default movement action
     rotate: ActionConfig = Field(default_factory=lambda: ActionConfig(enabled=False))
-    put_items: ActionConfig = Field(default_factory=lambda: ActionConfig(enabled=False))
-    place_box: ActionConfig = Field(default_factory=lambda: ActionConfig(enabled=False))
+    put_items: ActionConfig = Field(default_factory=lambda: ActionConfig(enabled=True))
     get_items: ActionConfig = Field(default_factory=lambda: ActionConfig(enabled=True))
     attack: AttackActionConfig = Field(default_factory=lambda: AttackActionConfig(enabled=False))
     swap: ActionConfig = Field(default_factory=lambda: ActionConfig(enabled=False))
@@ -107,14 +106,6 @@ class WallConfig(Config):
     swappable: bool = Field(default=False)
 
 
-class BoxConfig(Config):
-    """Python box configuration."""
-
-    type_id: int = Field(default=0, ge=0, le=255)
-    # We don't allow setting of returned_resources -- it should always match
-    # the consumed_resources by place_box.
-
-
 class ConverterConfig(Config):
     """Python converter configuration."""
 
@@ -160,7 +151,7 @@ class GameConfig(Config):
     agents: list[AgentConfig] = Field(default_factory=list)
     actions: ActionsConfig = Field(default_factory=lambda: ActionsConfig(noop=ActionConfig()))
     global_obs: GlobalObsConfig = Field(default_factory=GlobalObsConfig)
-    objects: dict[str, ConverterConfig | WallConfig | BoxConfig] = Field(default_factory=dict)
+    objects: dict[str, ConverterConfig | WallConfig] = Field(default_factory=dict)
     # these are not used in the C++ code, but we allow them to be set for other uses.
     # E.g., templates can use params as a place where values are expected to be written,
     # and other parts of the template can read from there.
@@ -180,6 +171,8 @@ class GameConfig(Config):
     )
     allow_diagonals: bool = Field(default=False, description="Enable actions to be aware of diagonal orientations")
 
+    reward_estimates: Optional[dict[str, float]] = Field(default=None)
+
 
 class MettaGridConfig(Config):
     """Environment configuration."""

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/metta/mettagrid/mettagrid_env.py

@@ -59,6 +59,8 @@ class MettaGridEnv(MettaGridPufferBase):
         self._episode_id: str | None = None
         self._last_reset_ts = datetime.datetime.now()
         self._is_training = is_training
+        self._label_completions = {"completed_tasks": [], "completion_rates": {}}
+        self.per_label_rewards = {}
 
         # DesyncEpisodes - when training we want to stagger experience. The first episode
         # will end early so that the next episode can begin at a different time on each worker.
@@ -125,7 +127,25 @@ class MettaGridEnv(MettaGridPufferBase):
         self.timer.start("thread_idle")
         return observations, rewards, terminals, truncations, infos
 
-    def _process_episode_completion(self, infos: Dict[str, Any]) -> None:
+    def _update_label_completions(self, moving_avg_window: int = 500) -> None:
+        """Update label completions."""
+        label = self.mg_config.label
+
+        # keep track of a list of the last 500 labels
+        if len(self._label_completions["completed_tasks"]) >= moving_avg_window:
+            self._label_completions["completed_tasks"].pop(0)
+        self._label_completions["completed_tasks"].append(label)
+
+        # moving average of the completion rates
+        self._label_completions["completion_rates"] = {t: 0 for t in set(self._label_completions["completed_tasks"])}
+        for t in self._label_completions["completed_tasks"]:
+            self._label_completions["completion_rates"][t] += 1
+        self._label_completions["completion_rates"] = {
+            t: self._label_completions["completion_rates"][t] / len(self._label_completions["completed_tasks"])
+            for t in self._label_completions["completion_rates"]
+        }
+
+    def _process_episode_completion(self, infos: Dict[str, Any], moving_avg_window: int = 500, alpha=0.9) -> None:
         """Process episode completion - stats, etc."""
         self.timer.start("process_episode_stats")
 
@@ -148,6 +168,29 @@ class MettaGridEnv(MettaGridPufferBase):
         for n, v in infos["agent"].items():
             infos["agent"][n] = v / self.num_agents
 
+        # If reward estimates are set, plot them compared to the mean reward
+        if self.mg_config.game.reward_estimates:
+            infos["reward_estimates"] = {}
+            infos["reward_estimates"]["diff_from_efficient_optimal"] = (
+                self.mg_config.game.reward_estimates["most_efficient_optimal_reward"] - episode_rewards.mean()
+            )
+            infos["reward_estimates"]["diff_from_inefficient_optimal"] = (
+                self.mg_config.game.reward_estimates["least_efficient_optimal_reward"] - episode_rewards.mean()
+            )
+
+        self._update_label_completions(moving_avg_window)
+
+        # only plot label completions once we have a full moving average window, to prevent initial bias
+        if len(self._label_completions["completed_tasks"]) == moving_avg_window:
+            infos["label_completions"] = self._label_completions["completion_rates"]
+
+        if self.mg_config.label not in self.per_label_rewards:
+            self.per_label_rewards[self.mg_config.label] = 0
+        self.per_label_rewards[self.mg_config.label] += episode_rewards.mean()
+        infos["per_label_rewards"] = (
+            alpha * self.per_label_rewards[self.mg_config.label] + (1 - alpha) * episode_rewards.mean()
+        )
+
         # Add attributes
         attributes: Dict[str, Any] = {
             "seed": self._current_seed,

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/metta/mettagrid/objects/agent.hpp

@@ -10,7 +10,6 @@
 #include "../stats_tracker.hpp"
 #include "agent_config.hpp"
 #include "constants.hpp"
-#include "objects/box.hpp"
 #include "types.hpp"
 
 class Agent : public GridObject {
@@ -44,7 +43,6 @@ public:
   GridLocation prev_location;
   std::string prev_action_name;
   unsigned int steps_without_motion;
-  Box* box;
 
   Agent(GridCoord r, GridCoord c, const AgentConfig& config)
       : group(config.group_id),
@@ -67,8 +65,7 @@ public:
         reward(nullptr),
         prev_location(r, c, GridLayer::AgentLayer),
         prev_action_name(""),
-        steps_without_motion(0),
-        box(nullptr) {
+        steps_without_motion(0) {
     populate_initial_inventory(config.initial_inventory);
     GridObject::init(config.type_id, config.type_name, GridLocation(r, c, GridLayer::AgentLayer));
   }

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/mettagrid.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mettagrid
-Version: 0.2
+Version: 0.2.0.1
 Summary: A fast grid-based open-ended MARL environment
 Author-email: David Bloomin <daveey@gmail.com>
 License-Expression: MIT
@@ -18,7 +18,7 @@ Requires-Dist: google-auth-oauthlib>=1.0.0
 Requires-Dist: gymnasium==0.29.1
 Requires-Dist: matplotlib>=3.10.3
 Requires-Dist: numpy<2,>=1.26.4
-Requires-Dist: omegaconf>=2.4.0.dev3
+Requires-Dist: omegaconf>=2.3.0
 Requires-Dist: duckdb>=1.3.0
 Requires-Dist: pettingzoo<1.25,>=1.24.1
 Requires-Dist: pufferlib<3.1.0,>=3.0.0

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/mettagrid.egg-info/SOURCES.txt

@@ -3,7 +3,6 @@ BUILD.bazel
 LICENSE
 MANIFEST.in
 MODULE.bazel
-MODULE.bazel.lock
 README.md
 bazel_build.py
 pyproject.toml
@@ -23,7 +22,6 @@ src/metta/mettagrid/hash.hpp
 src/metta/mettagrid/mettagrid_c.cpp
 src/metta/mettagrid/mettagrid_c.hpp
 src/metta/mettagrid/mettagrid_c.pyi
-src/metta/mettagrid/mettagrid_c.so
 src/metta/mettagrid/mettagrid_c_config.py
 src/metta/mettagrid/mettagrid_config.hpp
 src/metta/mettagrid/mettagrid_config.py
@@ -45,7 +43,6 @@ src/metta/mettagrid/actions/get_output.hpp
 src/metta/mettagrid/actions/move.hpp
 src/metta/mettagrid/actions/noop.hpp
 src/metta/mettagrid/actions/orientation.hpp
-src/metta/mettagrid/actions/place_box.hpp
 src/metta/mettagrid/actions/put_recipe_items.hpp
 src/metta/mettagrid/actions/rotate.hpp
 src/metta/mettagrid/actions/swap.hpp
@@ -111,7 +108,6 @@ src/metta/mettagrid/mapgen/utils/thumbnail.py
 src/metta/mettagrid/objects/__init__.py
 src/metta/mettagrid/objects/agent.hpp
 src/metta/mettagrid/objects/agent_config.hpp
-src/metta/mettagrid/objects/box.hpp
 src/metta/mettagrid/objects/constants.hpp
 src/metta/mettagrid/objects/converter.hpp
 src/metta/mettagrid/objects/converter_config.hpp

{mettagrid-0.2 → mettagrid-0.2.0.1}/src/mettagrid.egg-info/requires.txt

@@ -6,7 +6,7 @@ google-auth-oauthlib>=1.0.0
 gymnasium==0.29.1
 matplotlib>=3.10.3
 numpy<2,>=1.26.4
-omegaconf>=2.4.0.dev3
+omegaconf>=2.3.0
 duckdb>=1.3.0
 pettingzoo<1.25,>=1.24.1
 pufferlib<3.1.0,>=3.0.0

{mettagrid-0.2 → mettagrid-0.2.0.1}/tests/test_action_compatibility.py

@@ -110,21 +110,20 @@ class TestActionOrdering:
         # Action order should remain the same despite different config order
         assert action_names1 == action_names2, "Action order should be deterministic"
 
-        # Verify the expected order
-        assert action_names1 == ["get_items", "move", "noop", "rotate"]
+        # Verify the expected order (put_items is now enabled by default)
+        assert action_names1 == ["get_items", "move", "noop", "put_items", "rotate"]
 
     def test_action_indices_consistency(self, basic_config, simple_map):
         """Test that action indices remain consistent."""
         env = MettaGrid(from_mettagrid_config(basic_config), simple_map, 42)
         action_names = env.action_names()
 
-        # Verify indices
+        # Verify indices (put_items is now enabled by default)
+        assert action_names.index("get_items") == 0
         assert action_names.index("move") == 1
         assert action_names.index("noop") == 2
-        assert action_names.index("rotate") == 3
-
-        # get_items is always present even if not explicitly configured
-        assert action_names.index("get_items") == 0
+        assert action_names.index("put_items") == 3
+        assert action_names.index("rotate") == 4
 
 
 class TestActionValidation:
@@ -394,8 +393,8 @@ class TestSpecialActions:
         # Attack should be present
         assert "attack" in action_names
 
-        # Check the expected order (attack comes before other actions)
-        expected_actions = ["attack", "get_items", "move", "noop", "rotate"]
+        # Check the expected order (attack comes before other actions, put_items is enabled by default)
+        expected_actions = ["attack", "get_items", "move", "noop", "put_items", "rotate"]
         assert action_names == expected_actions
 
     def test_swap_action_registration(self, basic_config, simple_map):