added dev environment, fix polars native expr.

projectmesa · Jul 9, 2024 · f6413b1 · f6413b1
1 parent bf1ccd9
commit f6413b1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/docs/scripts/readme_plot.py b/docs/scripts/readme_plot.py
@@ -129,11 +129,9 @@ def give_money(self):
 
         # Add the income to the other agents
         # 1. Using native expressions
-        """self.agents = self.agents.with_columns(
-            pl.when(pl.col("unique_id").is_in(new_wealth["unique_id"]))
-            .then(pl.col("wealth") + new_wealth["wealth"])
-            .otherwise(pl.col("wealth"))
-        )"""
+        """self.agents = self.agents.join(new_wealth, on="unique_id", how="left").fill_null(
+            0
+        ).with_columns(wealth=pl.col("wealth") + pl.col("len")).drop("len")"""
 
         # 2. Using the set method
         """self.set(

diff --git a/pyproject.toml b/pyproject.toml
@@ -31,5 +31,8 @@ dev = [
   "mesa",
 ]
 
+[tool.hatch.envs.dev] #Allows installing dev as virtual env
+features = ["dev"]
+
 [tool.hatch.build.targets.wheel]
 packages = ["mesa_frames"]