Merge pull request #58 from personalrobotics/fix/src-layout

Migrate to src/ layout

Author: siddhss5

pyproject.toml

@@ -27,8 +27,11 @@ test = [
 asset-manager = { workspace = true }
 
 [build-system]
-requires = ["setuptools>=61.0"]
-build-backend = "setuptools.build_meta"
+requires = ["hatchling"]
+build-backend = "hatchling.build"
 
-[tool.setuptools]
-packages = ["mj_environment"]
+[tool.hatch.metadata]
+allow-direct-references = true
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/mj_environment"]

mj_environment/__init__.py src/mj_environment/__init__.pymj_environment/__init__.py renamed to src/mj_environment/__init__.py

@@ -3,6 +3,8 @@
 Manages object lifecycle (activation, hiding, movement) in MuJoCo.
 """
 
+from __future__ import annotations
+
 import logging
 import math
 from typing import Any, Dict, List, NamedTuple, Optional, Sequence, Union
@@ -119,8 +121,8 @@ def __init__(
             scene_cfg: Objects dict from scene_config.yaml (already loaded by Environment)
             hide_pos: Position for hidden objects
         """
-        self.model = model
-        self.data = data
+        self.model: mujoco.MjModel = model
+        self.data: mujoco.MjData = data
         self.asset_manager = asset_manager
         self.hide_pos = np.array(hide_pos, dtype=float)
         self.objects: Dict[str, Dict[str, Any]] = {}
@@ -345,7 +347,7 @@ def activate(
         logger.debug("Activated %s", name)
         return name
 
-    def hide(self, name: str):
+    def hide(self, name: str) -> None:
         """Hide an active object by moving and disabling it."""
         if name not in self.active_objects:
             raise ObjectNotFoundError(name, list(self.active_objects.keys()))