feat: add build system, mypy and editable source

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