feat: add build system, mypy and editable source

pyproject.toml

@@ -1,3 +1,7 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
 [project]
 name = "2026sel3-project"
 version = "0.1.0"
@@ -41,3 +45,36 @@ dev = [
     "pytest>=8.0.0",
     "ruff>=0.15.2",
 ]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/brittle_star_project", "src/experiment_logger"]
+
+[tool.mypy]
+mypy_path = "src"
+check_untyped_defs = false
+warn_return_any = false
+
+[[tool.mypy.overrides]]
+module = [
+    "jax.*",
+    "flax.*",
+    "wandb.*",
+    "torch.*",
+    "mujoco.*",
+    "mujoco_warp.*",
+    "optax.*",
+    "tyro.*",
+    "biorobot.*",
+    "gymnasium.*",
+    "matplotlib.*",
+    "mediapy.*",
+    "matplotlib.*",
+    "mediapy.*",
+    "pytest.*",
+    "tensorboard.*",
+    "tqdm.*",
+    "numpy.*",
+    "yaml.*",
+    "moojoco.*"
+]
+ignore_missing_imports = true

src/__init__.py

@@ -1,25 +0,0 @@
-from .brittle_star_project import (
-    ArenaConfig,
-    Backend,
-    BrittleStarEnv,
-    BrittleStarEnvFactory,
-    EnvConfig,
-    MorphologyConfig,
-    Task,
-    simulate_policy,
-    SimulationConfig,
-    ControlPolicy,
-)
-
-__all__ = [
-    "ArenaConfig",
-    "Backend",
-    "BrittleStarEnv",
-    "BrittleStarEnvFactory",
-    "EnvConfig",
-    "MorphologyConfig",
-    "Task",
-    "simulate_policy",
-    "SimulationConfig",
-    "ControlPolicy",
-]

src/brittle_star_project/trainers/PPOTrainer.py

@@ -16,14 +16,14 @@ from experiment_logger import get_logger
 
 from brittle_star_project.dataclasses import EpisodeStatistics, PPOArgs
 from brittle_star_project.environment.BrittleStarJaxEnvWrapper import BrittleStarJaxEnvWrapper
-from MLPs.mlps import (
+from brittle_star_project.MLPs.mlps import (
     Actor,
     AgentParams,
     GenericDenseLayersWithActivation,
     OneDenseLayerMLP,
     Storage,
 )
-from ppo import PPO
+from brittle_star_project.ppo import PPO
 
 
 @jax.jit

uv.lock

@@ -11,7 +11,7 @@ resolution-markers = [
 [[package]]
 name = "2026sel3-project"
 version = "0.1.0"
-source = { virtual = "." }
+source = { editable = "." }
 dependencies = [
     { name = "biorobot" },
     { name = "cleanrl" },