Multi-Agent-Water-Resources-Management [SarmadZandi].ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": []
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "h6sVCV9e6odk",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "45f4c971-0564-4ba6-e39a-6d4c60e1eace"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Collecting pettingzoo\n",
            "  Downloading pettingzoo-1.24.3-py3-none-any.whl (847 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m847.8/847.8 kB\u001b[0m \u001b[31m7.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: numpy>=1.21.0 in /usr/local/lib/python3.10/dist-packages (from pettingzoo) (1.25.2)\n",
            "Collecting gymnasium>=0.28.0 (from pettingzoo)\n",
            "  Downloading gymnasium-0.29.1-py3-none-any.whl (953 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m953.9/953.9 kB\u001b[0m \u001b[31m12.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: cloudpickle>=1.2.0 in /usr/local/lib/python3.10/dist-packages (from gymnasium>=0.28.0->pettingzoo) (2.2.1)\n",
            "Requirement already satisfied: typing-extensions>=4.3.0 in /usr/local/lib/python3.10/dist-packages (from gymnasium>=0.28.0->pettingzoo) (4.11.0)\n",
            "Collecting farama-notifications>=0.0.1 (from gymnasium>=0.28.0->pettingzoo)\n",
            "  Downloading Farama_Notifications-0.0.4-py3-none-any.whl (2.5 kB)\n",
            "Installing collected packages: farama-notifications, gymnasium, pettingzoo\n",
            "Successfully installed farama-notifications-0.0.4 gymnasium-0.29.1 pettingzoo-1.24.3\n"
          ]
        }
      ],
      "source": [
        "\n",
        "!pip install pettingzoo"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "import pettingzoo\n",
        "from pettingzoo import ParallelEnv\n",
        "from pettingzoo import AECEnv\n",
        "from pettingzoo.utils import parallel_to_aec, wrappers, agent_selector\n",
        "from pettingzoo.utils import ClipOutOfBoundsWrapper\n",
        "from pettingzoo.utils.conversions import parallel_wrapper_fn\n",
        "from pettingzoo.sisl._utils import Agent\n",
        "import gymnasium\n",
        "from gymnasium.utils import EzPickle ,seeding\n",
        "from gymnasium.spaces import Discrete, MultiDiscrete, Dict, Box\n",
        "from gymnasium import spaces\n",
        "import random\n",
        "import functools\n",
        "import numpy as np\n",
        "import pandas as pd\n",
        "from copy import copy\n",
        "from math import pi, sin\n",
        "from random import gauss, uniform\n",
        "from tabulate import tabulate\n",
        "import matplotlib.pyplot as plt\n",
        "import importlib\n",
        "\n",
        "def check_library_versions(libraries):\n",
        "  print(\"The version of each installed library:\")\n",
        "  print(\"{:<30} {:<15}\".format(\"Library\", \"Version\"))\n",
        "  print(\"-\" * 45)\n",
        "  for lib in libraries:\n",
        "    try:\n",
        "      module = importlib.import_module(lib)\n",
        "      version = getattr(module, '__version__', 'Version not available')\n",
        "      print(\"{:<30} {:<15}\".format(lib, version))\n",
        "    except ImportError:\n",
        "      print(\"{:<30} {:<15}\".format(lib, \"Failed to import\"))\n",
        "  print(\"-\" * 45)\n",
        "\n",
        "libraries = [\n",
        "    \"pettingzoo\", \"gymnasium\", \"numpy\", \"pandas\"\n",
        "]\n",
        "\n",
        "# Check the version of each library\n",
        "check_library_versions(libraries)\n"
      ],
      "metadata": {
        "id": "ii1SUK1AG3L_",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "a822df4d-2629-419c-f0ba-7f77adb0c540"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "The version of each installed library:\n",
            "Library                        Version        \n",
            "---------------------------------------------\n",
            "pettingzoo                     1.24.3         \n",
            "gymnasium                      0.29.1         \n",
            "numpy                          1.25.2         \n",
            "pandas                         2.0.3          \n",
            "---------------------------------------------\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "#----------------------------------------------------------------------------------------------------------\n",
        "# Custom environment class extending PettingZoo's AECEnv\n",
        "class TehranWaterManagementEnv(AECEnv):\n",
        "    metadata = {'render.modes': ['human']}\n",
        "\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def __init__(self):\n",
        "        super().__init__()\n",
        "\n",
        "        # Initialize environment state\n",
        "        # priority handling\n",
        "        self.agents = [\"industry_1\", \"industry_2\", \"industry_3\", \"municipal_1\", \"municipal_2\", \"municipal_3\", \"municipal_4\", \"municipal_5\", \"farmer_1\", \"farmer_2\"]\n",
        "        self.agent_name_mapping = dict(zip(self.agents, list(range(len(self.agents)))))\n",
        "        self._agent_selector = agent_selector(self.agents)\n",
        "        self.agent_selection = self._agent_selector.next()\n",
        "\n",
        "        # Groundwater Aquifer Initialization\n",
        "        self.groundwater_levels = np.array([1174, 1058, 1001])  # Initial water levels for areas 1, 2, 3\n",
        "        self.nitrate_density = np.array([1, 1, 1])  # Initial nitrate density for each area\n",
        "        self.area_thresholds = np.array([89, 86, 38])  # Thresholds for each area\n",
        "        self.incoming_water_flow = self._random_incoming_water_flow()  # Initialize random incoming water flow\n",
        "        self.withdrawal_limits = {\"farmer_1\": 50e6, \"farmer_2\": 50e6,\n",
        "                                  \"industry_1\": 0.19 * 20.09e6, \"industry_2\": 0.69 * 20.09e6, \"industry_3\": 0.12 * 20.09e6,\n",
        "                                  \"municipal_1\": 40, \"municipal_2\": 40, \"municipal_3\": 40, \"municipal_4\": 40, \"municipal_5\": 40}  # Example limits\n",
        "\n",
        "        # Define observation and action spaces for all agents\n",
        "        self.action_spaces = {agent: self._define_action_space(agent) for agent in self.agents}\n",
        "        self.observation_spaces = {agent: self._define_observation_space(agent) for agent in self.agents}\n",
        "\n",
        "        # Environment Parameters & States\n",
        "        self.current_step = 0\n",
        "        self.max_steps = 25\n",
        "        self.terminate = False\n",
        "        self.rewards = {agent: 0 for agent in self.agents}\n",
        "        self.illegal_withdrawal_fines = {agent: 0 for agent in self.agents}\n",
        "        self.actual_water_supply_percentage = {agent: 100 for agent in self.agents}  # Assuming 100% needs met initially\n",
        "        self.water_demand = {agent: 0 for agent in self.agents}\n",
        "\n",
        "        # Crop Information & Economic Parameters\n",
        "        self.crop_info = {\"wheat\": {\"PPC\": 12705, \"CPD\": 0.63, \"Eff\": 5352},\n",
        "                          \"barley\": {\"PPC\": 10028, \"CPD\": 0.70, \"Eff\": 3840},\n",
        "                          \"alfalfa\": {\"PPC\": 7700, \"CPD\": 1.09, \"Eff\": 12000}}\n",
        "\n",
        "        self.P_OT = 75000  # Price of water per cubic meter (Rials)\n",
        "        self.PE_1 = 30000  # Price of electricity per kWh (Rials)\n",
        "\n",
        "        # Municipal Green Space Parameters\n",
        "        self.potential_green_space = {\"municipal_1\": 200000, \"municipal_2\": 150000, \"municipal_3\": 180000, \"municipal_4\": 160000, \"municipal_5\": 140000}\n",
        "        self.current_green_space = {\"municipal_1\": 100000, \"municipal_2\": 75000, \"municipal_3\": 90000, \"municipal_4\": 80000, \"municipal_5\": 70000}\n",
        "\n",
        "        # Water Allocation & Legal Withdrawal Limits\n",
        "        self.water_allocation = self._init_water_allocation()\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def _random_incoming_water_flow(self):\n",
        "        # Generates random incoming water flow for each area of the aquifer\n",
        "        return np.random.uniform(0.1, 10, size=3)\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def _define_action_space(self, agent):\n",
        "        if \"farmer\" in agent:\n",
        "            # Example: 3 actions for crop selection, 1 for water needed, 1 for illegal withdrawal\n",
        "            # Assuming discrete actions for simplicity\n",
        "            return spaces.MultiDiscrete([3, 100, 1])  # 3 crops, 0-100% water, 0 or 1 for illegal withdrawal\n",
        "        elif \"industry\" in agent:\n",
        "            # Example: 1 action for illegal withdrawal\n",
        "            return spaces.Discrete(2)  # 0 or 1 for illegal withdrawal\n",
        "        elif \"municipal\" in agent:\n",
        "            # Example: 1 for water needed, 1 for illegal withdrawal, 1 for green space expansion\n",
        "            return spaces.MultiDiscrete([100, 1, 100])  # 0-100% water needed, 0 or 1 illegal, 0-100% green space expansion\n",
        "        else:\n",
        "            raise NotImplementedError(f\"Unknown agent type: {agent}\")\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def _define_observation_space(self, agent):\n",
        "        groundwater_level_bins = 500  # Example: 500 discrete levels\n",
        "        # Additional observations can be added as needed, adjust according to your scenario.\n",
        "        if \"farmer\" in agent or \"industry\" in agent or \"municipal\" in agent:\n",
        "            # Assuming the only observation for simplicity is the groundwater level.\n",
        "            # If there are other observations, consider using MultiDiscrete or a similar approach.\n",
        "            return spaces.Discrete(groundwater_level_bins)\n",
        "        else:\n",
        "            raise NotImplementedError(f\"Unknown agent type: {agent}\")\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def _init_water_allocation(self):\n",
        "        water_allocation = {\n",
        "            \"farmer_1\": {\"allocated_water\": 100, \"legal_withdrawal_limit\": 50},  # Example values\n",
        "            \"farmer_2\": {\"allocated_water\": 80, \"legal_withdrawal_limit\": 50},\n",
        "            \"industry_1\": {\"allocated_water\": 380, \"legal_withdrawal_limit\": 0.4 * 380},  # 19% of total, 40% withdrawal\n",
        "            \"industry_2\": {\"allocated_water\": 1382, \"legal_withdrawal_limit\": 0.4 * 1382},  # 69% of total, 40% withdrawal\n",
        "            \"industry_3\": {\"allocated_water\": 238, \"legal_withdrawal_limit\": 0.4 * 238},  # 12% of total, 40% withdrawal\n",
        "            \"municipal_1\": {\"allocated_water\": 200, \"legal_withdrawal_limit\": 0.4 * 200},  # Example values\n",
        "            \"municipal_2\": {\"allocated_water\": 200, \"legal_withdrawal_limit\": 0.4 * 200},\n",
        "            \"municipal_3\": {\"allocated_water\": 200, \"legal_withdrawal_limit\": 0.4 * 200},\n",
        "            \"municipal_4\": {\"allocated_water\": 200, \"legal_withdrawal_limit\": 0.4 * 200},\n",
        "            \"municipal_5\": {\"allocated_water\": 200, \"legal_withdrawal_limit\": 0.4 * 200}\n",
        "        }\n",
        "        return water_allocation\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def update_aquifer_state(self, actions):\n",
        "        # Constants and area parameters\n",
        "        area = np.array([78_125_000, 242_750_000, 133_187_500])  # Area of regions 1, 2, and 3\n",
        "        specific_yield = np.array([0.15, 0.15, 0.15])  # Assuming a constant specific yield for simplicity\n",
        "        length = np.array([7500, 16000, 12000])  # Length of each region\n",
        "        hydraulic_conductivity = np.array([0.002, 0.002, 0.002])  # Assuming a constant value for simplicity\n",
        "        b = np.array([14250, 15750])  # Width between regions\n",
        "        S = specific_yield  # Specific yield, using the same value for simplicity\n",
        "\n",
        "        # Update incoming water flow for the new timestep\n",
        "        self.incoming_water_flow = self._random_incoming_water_flow()\n",
        "\n",
        "        # Calculate Di_T, Re_T, and Tr for each area\n",
        "        Di_T = np.zeros(3)  # Total water extracted from each region\n",
        "        Re_T = np.zeros(3)  # Total recharge in each region\n",
        "        Tr = np.zeros(2)    # Water transfer between regions\n",
        "\n",
        "\n",
        "        Re_T = np.array([1e5, 2e5, 1.5e5])  # Just for example purposes\n",
        "        Di_T = np.array([1e4, 1e4, 1e4])    # Example withdrawal, should be derived from `actions`\n",
        "\n",
        "        # Calculating water transfer (Tr) based on the formula provided\n",
        "        for i in range(2):\n",
        "            Tr[i] = 365 * ((self.groundwater_levels[i] - self.groundwater_levels[i + 1]) / ((length[i] + length[i + 1]) / 2)) * (length[i] + length[i + 1]) / ((length[i] / hydraulic_conductivity[i]) + (length[i + 1] / hydraulic_conductivity[i + 1])) * b[i] * ((self.groundwater_levels[i] + self.groundwater_levels[i + 1]) / 2)\n",
        "\n",
        "        # Update groundwater levels (B_(n,i+1)) for each area based on water balance equation\n",
        "        for n in range(3):\n",
        "            An = area[n]\n",
        "            B_n_i = self.groundwater_levels[n]  # Current groundwater level for area n\n",
        "            Tr_in = Tr[n - 1] if n > 0 else 0  # Incoming transfer (none for the first area)\n",
        "            Tr_out = Tr[n] if n < 2 else 0    # Outgoing transfer (none for the last area)\n",
        "            Recharge = Re_T[n]  # Total recharge for area n\n",
        "            Withdrawal = Di_T[n]  # Total withdrawal from area n\n",
        "\n",
        "            # Water balance equation to calculate the new groundwater level\n",
        "            B_n_i_plus_1 = B_n_i + (Recharge - Withdrawal + Tr_in - Tr_out) / (S[n] * An)\n",
        "            self.groundwater_levels[n] = max(B_n_i_plus_1, self.area_thresholds[n])  # Ensuring not falling below threshold\n",
        "\n",
        "        # Threshold check and adjustments\n",
        "        for n, level in enumerate(self.groundwater_levels):\n",
        "            if level < [89, 86, 38][n]:  # Threshold levels for each area\n",
        "                print(f\"Warning: Water level in Area {n+1} below threshold!\")\n",
        "        print(\"Updated Groundwater levels: \",self.groundwater_levels)\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def step(self, action):\n",
        "        # Store the last action for rendering purposes\n",
        "        self.last_action = action\n",
        "\n",
        "        # Check the agent type and handle the action accordingly\n",
        "        if self.agent_selection.startswith(\"farmer\"):\n",
        "            self._handle_farmer_action(action)\n",
        "            # Assuming action structure for farmers is (crop_allocation, requested_water, illegal_withdrawal)\n",
        "        elif self.agent_selection.startswith(\"industry\"):\n",
        "            self._handle_industry_action(action)\n",
        "            # Industry actions do not have a second element for water requested, handle accordingly\n",
        "        elif self.agent_selection.startswith(\"municipal\"):\n",
        "            self._handle_municipal_action(action)\n",
        "\n",
        "        # Update aquifer state after handling the action\n",
        "        self.update_aquifer_state(action)\n",
        "\n",
        "        # Update rewards and prepare for the next step\n",
        "        self.agent_selection = self._agent_selector.next()\n",
        "        self.current_step += 1  # Move to the next step\n",
        "\n",
        "        # Check for termination condition\n",
        "        self._check_termination_conditions()\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def _handle_farmer_action(self, action):\n",
        "        # Extract action components for a farmer\n",
        "        crop_percentages, requested_water, illegal_withdrawal = action\n",
        "\n",
        "        # Identifying the farmer's area based on the agent_selection\n",
        "        area_idx = 1 if self.agent_selection == \"farmer_1\" else 2  # Adjusting based on your setup\n",
        "\n",
        "        # Calculate the water demand (DF) for the farmer's crops\n",
        "        area_hectares = 5103.22 if area_idx == 1 else 3127.78  # Area in hectares for farmer 1 and 2\n",
        "        water_demand = 0  # Initialize water demand\n",
        "\n",
        "        for i, crop in enumerate([\"wheat\", \"barley\", \"alfalfa\"]):\n",
        "            crop_info = self.crop_info[crop]\n",
        "            crop_demand = (crop_percentages[i] / 100.0) * area_hectares * crop_info['Eff'] / crop_info['CPD']\n",
        "            water_demand += crop_demand\n",
        "\n",
        "        # Calculate the actual water available (AF) for the farmer\n",
        "        allocated_water = self.water_allocation[self.agent_selection][\"allocated_water\"]\n",
        "        legal_withdrawal_limit = self.water_allocation[self.agent_selection][\"legal_withdrawal_limit\"]\n",
        "        actual_water_available = allocated_water + (legal_withdrawal_limit if not illegal_withdrawal else requested_water)\n",
        "        actual_water_available = min(actual_water_available, self.groundwater_levels[area_idx])  # Ensure it doesn't exceed the groundwater level\n",
        "\n",
        "        # Update groundwater level based on the actual water used\n",
        "        self.groundwater_levels[area_idx] -= actual_water_available\n",
        "\n",
        "        # Calculate the reward based on the farmer's action and update water demand\n",
        "        if water_demand > actual_water_available and illegal_withdrawal:\n",
        "            self.rewards[self.agent_selection] = -1\n",
        "        else:\n",
        "            # Simplify profit calculation for demonstration purposes\n",
        "            profit = actual_water_available * sum(\n",
        "                (crop_percentages[i] / 100.0) * self.crop_info[crop]['PPC'] * self.crop_info[crop]['CPD']\n",
        "                for i, crop in enumerate([\"wheat\", \"barley\", \"alfalfa\"])\n",
        "            )\n",
        "            # Calculate reward based on profit, adjusting for water price and energy cost\n",
        "            self.rewards[self.agent_selection] = profit / (self.P_OT + self.PE_1 * self.groundwater_levels[area_idx])\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def _handle_industry_action(self, action):\n",
        "        # Illegal withdrawal is a boolean: 0 or 1\n",
        "        illegal_withdrawal = action\n",
        "\n",
        "        # Industry water needs (percentage of total industrial water need)\n",
        "        industry_water_needs = {\n",
        "            \"industry_1\": 0.19 * 20.09,  # 19%\n",
        "            \"industry_2\": 0.69 * 20.09,  # 69%\n",
        "            \"industry_3\": 0.12 * 20.09,  # 12%\n",
        "        }\n",
        "\n",
        "        # Calculating the demand (DW) and allocated water (AW) for the industry\n",
        "        DW = industry_water_needs[self.agent_selection]  # Demand in million cubic meters\n",
        "        AW = self.water_allocation[self.agent_selection][\"allocated_water\"]  # Allocated water from the initial setup\n",
        "\n",
        "        # Actual water that can be legally withdrawn, based on the agent's allocation and the environment\n",
        "        legal_withdrawal_limit = self.water_allocation[self.agent_selection][\"legal_withdrawal_limit\"]\n",
        "        actual_withdrawal_capacity = AW + legal_withdrawal_limit\n",
        "\n",
        "        # Determine if the demand exceeds what is legally available, including potential illegal withdrawal\n",
        "        exceeds_demand = DW > actual_withdrawal_capacity\n",
        "\n",
        "        # If illegal withdrawal is made and demand exceeds the legal limit\n",
        "        if illegal_withdrawal and exceeds_demand:\n",
        "            self.rewards[self.agent_selection] = -1  # Penalize for illegal action\n",
        "        else:\n",
        "            # Calculate the reward based on the difference between demand and the water supplied\n",
        "            # Ensure that actual_withdrawal does not exceed the DW even when illegal withdrawal is considered\n",
        "            actual_withdrawal = min(DW, actual_withdrawal_capacity) if not illegal_withdrawal else min(DW, AW + 0.4 * DW)\n",
        "            self.rewards[self.agent_selection] = 1 - abs(DW - actual_withdrawal) / DW\n",
        "\n",
        "        # Update the groundwater level based on actual withdrawal\n",
        "        # Assuming an industry mapping to area indices: Industry 1 -> Area 1, etc.\n",
        "        area_idx = int(self.agent_selection.split(\"_\")[1]) - 1\n",
        "        self.groundwater_levels[area_idx] -= actual_withdrawal\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def _handle_municipal_action(self, action):\n",
        "        # Unpack the action components for clarity\n",
        "        water_needed, illegal_withdrawal, green_space_expansion_percentage = action\n",
        "        agent_region = self.agent_selection\n",
        "\n",
        "        # Assuming regions 5, 22 are in Area 1, regions 9, 18 in Area 2, and region 21 partially in Areas 1 and 2\n",
        "        area_index_mapping = {\"municipal_1\": 0, \"municipal_2\": 1, \"municipal_3\": 1, \"municipal_4\": 0, \"municipal_5\": 0}\n",
        "        area_index = area_index_mapping[agent_region]\n",
        "\n",
        "        # Calculate the maximum possible green space expansion for this step\n",
        "        Delta_G_max = (self.potential_green_space[agent_region] - self.current_green_space[agent_region]) / 25\n",
        "        Area_G_current = self.current_green_space[agent_region] + green_space_expansion_percentage * Delta_G_max\n",
        "\n",
        "        # Calculate the total water demand for the green space\n",
        "        DG = Area_G_current * 1.5  # Assuming 1.5 cubic meters of water per square meter of green space\n",
        "\n",
        "        # Calculate the actual water available for the municipal agent\n",
        "        allocated_water = self.water_allocation[agent_region][\"allocated_water\"]\n",
        "        legal_withdrawal_limit = self.water_allocation[agent_region][\"legal_withdrawal_limit\"]\n",
        "        actual_water_available = min(allocated_water + legal_withdrawal_limit, self.groundwater_levels[area_index])\n",
        "\n",
        "        # Determine the reward based on the water demand and the actual water available\n",
        "        if DG > actual_water_available and illegal_withdrawal:\n",
        "            self.rewards[agent_region] = -1\n",
        "        else:\n",
        "            AG = min(DG, actual_water_available)  # The actual water used is the lesser of the demand or the available\n",
        "            reward_factor = (1 - abs(DG - AG) / DG)\n",
        "            green_space_factor = (Area_G_current - self.current_green_space[agent_region]) / (self.potential_green_space[agent_region] - self.current_green_space[agent_region])\n",
        "            self.rewards[agent_region] = reward_factor * green_space_factor * 100\n",
        "\n",
        "        # Update the groundwater level based on actual water used\n",
        "        self.groundwater_levels[area_index] -= AG if illegal_withdrawal else 0\n",
        "        self.current_green_space[agent_region] = Area_G_current\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def _check_termination_conditions(self):\n",
        "        # Check for groundwater levels below minimum thresholds\n",
        "        bedrock_levels = np.array([1077, 962, 959])\n",
        "        min_levels_above_bedrock = np.array([89, 86, 38])\n",
        "        water_levels_above_bedrock = self.groundwater_levels - bedrock_levels\n",
        "\n",
        "        if any(water_levels_above_bedrock < min_levels_above_bedrock):\n",
        "            self.terminate = True\n",
        "            # Apply penalty to all agents for groundwater level falling below threshold\n",
        "            for agent in self.agents:\n",
        "                self.rewards[agent] -= 1\n",
        "\n",
        "        # Check if max steps have been reached\n",
        "        if self.current_step >= self.max_steps:\n",
        "            self.terminate = True\n",
        "\n",
        "\n",
        "        # Placeholder: Terminate after one round of actions for testing\n",
        "        # In a real scenario, this should check for the end of a 25-year period or other termination conditions\n",
        "        self.terminated = True\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def reset(self):\n",
        "        # Reinitialize environment state to start a new episode\n",
        "\n",
        "        # Reset groundwater levels to initial values\n",
        "        self.groundwater_levels = np.array([1174, 1058, 1001])\n",
        "\n",
        "        # Reset water allocation based on initial settings\n",
        "        self.water_allocation = self._init_water_allocation()\n",
        "\n",
        "        # Reset rewards for all agents\n",
        "        self.rewards = {agent: 0 for agent in self.agents}\n",
        "\n",
        "        # Reset the nitrate density to initial values\n",
        "        self.nitrate_density = [1, 1, 1]\n",
        "\n",
        "        # Reset actual water supply percentages to 100%\n",
        "        self.actual_water_supply_percentage = {agent: 100 for agent in self.agents}\n",
        "\n",
        "        # Reset tracking of water demand for each agent\n",
        "        self.water_demand = {agent: 0 for agent in self.agents}\n",
        "\n",
        "        # Reset current step and termination flag\n",
        "        self.current_step = 0\n",
        "        self.terminate = False\n",
        "\n",
        "        # Reset current green space area to initial values\n",
        "        self.current_green_space = {\n",
        "            \"municipal_1\": 100000,\n",
        "            \"municipal_2\": 75000,\n",
        "            \"municipal_3\": 90000,\n",
        "            \"municipal_4\": 80000,\n",
        "            \"municipal_5\": 70000,\n",
        "        }\n",
        "\n",
        "        # Reset agent selection for the start of the episode\n",
        "        self._agent_selector = agent_selector(self.agents)\n",
        "        self.agent_selection = self._agent_selector.next()\n",
        "\n",
        "        # This aligns with practices in some RL environments where the reset method also provides the initial state observation\n",
        "        return self.observe(self.agent_selection)\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def render(self, mode='human'):\n",
        "        if mode == 'human':\n",
        "            print(\"\\nCurrent Environment step:\", self.current_step)\n",
        "\n",
        "            # Groundwater levels for each area\n",
        "            groundwater_table = []\n",
        "            for i, level in enumerate(self.groundwater_levels):\n",
        "                groundwater_table.append([f\"Area {i+1}\", f\"{level:.2f} million cubic meters\"])\n",
        "\n",
        "            print(\"\\nGroundwater Levels:\")\n",
        "            print(tabulate(groundwater_table, headers=[\"Area\", \"Groundwater Level\"], tablefmt=\"rounded_outline\"))\n",
        "\n",
        "            # Nitrate density for each area\n",
        "            nitrate_table = []\n",
        "            for i, density in enumerate(self.nitrate_density):\n",
        "                nitrate_table.append([f\"Area {i+1}\", f\"{density:.2f} mg/L\"])\n",
        "\n",
        "            print(\"\\nNitrate Density:\")\n",
        "            print(tabulate(nitrate_table, headers=[\"Area\", \"Nitrate Density\"], tablefmt=\"rounded_outline\"))\n",
        "\n",
        "            # Water allocation and demand for each agent\n",
        "            water_allocation_table = []\n",
        "            for agent, allocation in self.water_allocation.items():\n",
        "                water_allocation_table.append([agent, f\"{allocation['allocated_water']:.2f} million cubic meters\", f\"{allocation['legal_withdrawal_limit']:.2f} million cubic meters\"])\n",
        "\n",
        "            print(\"\\nWater Allocation and Usage:\")\n",
        "            print(tabulate(water_allocation_table, headers=[\"Agent\", \"Allocated Water (million cubic meters)\", \"Legal Withdrawal Limit (million cubic meters)\"], tablefmt=\"rounded_outline\"))\n",
        "\n",
        "            # Current green space for municipal agents\n",
        "            green_space_table = []\n",
        "            for agent, area in self.current_green_space.items():\n",
        "                if agent.startswith(\"municipal\"):\n",
        "                    green_space_table.append([agent, f\"{area:.2f} square meters\"])\n",
        "\n",
        "            print(\"\\nGreen Spaces Managed by Municipal Agents:\")\n",
        "            print(tabulate(green_space_table, headers=[\"Agent\", \"Current Green Space Area\"], tablefmt=\"rounded_outline\"))\n",
        "\n",
        "            # Rewards\n",
        "            rewards_table = []\n",
        "            for agent, reward in self.rewards.items():\n",
        "                rewards_table.append([agent, f\"{reward:.2f}\"])\n",
        "\n",
        "            print(\"\\nRewards:\")\n",
        "            print(tabulate(rewards_table, headers=[\"Agent\", \"Reward\"], tablefmt=\"rounded_outline\"))\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "    def observe(self, agent):\n",
        "        # Groundwater levels per area\n",
        "        groundwater_observation = np.array(self.groundwater_levels)\n",
        "\n",
        "        return groundwater_observation\n",
        "#----------------------------------------------------------------------------------------------------------\n",
        "# Initialize the environment\n",
        "env = TehranWaterManagementEnv()\n",
        "\n",
        "env.reset()\n",
        "# Run a test step for each agent\n",
        "# Revised test loop with appropriate default actions for each agent type\n",
        "for agent in env.agents:\n",
        "    print(\"=======================================================\")\n",
        "    print(f\"                  Testing {agent}\")\n",
        "    print(\"=======================================================\")\n",
        "    observation_before = env.observe(agent)\n",
        "    print(f\"Initial observation for {agent}: {observation_before}\")\n",
        "\n",
        "    # Provide default actions based on agent type\n",
        "    if \"farmer\" in agent:\n",
        "        action = ([50, 30, 20], 25, 0)  # Assuming 50% wheat, 30% barley, 20% alfalfa, 25 million cubic meters requested, no illegal withdrawal\n",
        "    elif \"industry\" in agent:\n",
        "        action = (1,)  # Assuming an example action for industry agents\n",
        "    elif \"municipal\" in agent:\n",
        "        action = (15, 0, 5)  # Assuming an example action for municipal agents\n",
        "\n",
        "    print(\"Action: \", action)\n",
        "    env.step(action)\n",
        "    observation_after = env.observe(agent)\n",
        "    print(f\"Observation after action for {agent}: {observation_after}\")\n",
        "    env.render()\n",
        "    print(\"----------------------------------------------------------------------------------------------\\n\")\n",
        "#----------------------------------------------------------------------------------------------------------"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "t_HmliqMGxSk",
        "outputId": "a85fff0b-12ce-48cd-91d8-293920307b04"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "=======================================================\n",
            "                  Testing industry_1\n",
            "=======================================================\n",
            "Initial observation for industry_1: [1174 1058 1001]\n",
            "Action:  (1,)\n",
            "Updated Groundwater levels:  [1169 1058 1001]\n",
            "Observation after action for industry_1: [1169 1058 1001]\n",
            "\n",
            "Current Environment step: 1\n",
            "\n",
            "Groundwater Levels:\n",
            "╭────────┬──────────────────────────────╮\n",
            "│ Area   │ Groundwater Level            │\n",
            "├────────┼──────────────────────────────┤\n",
            "│ Area 1 │ 1169.00 million cubic meters │\n",
            "│ Area 2 │ 1058.00 million cubic meters │\n",
            "│ Area 3 │ 1001.00 million cubic meters │\n",
            "╰────────┴──────────────────────────────╯\n",
            "\n",
            "Nitrate Density:\n",
            "╭────────┬───────────────────╮\n",
            "│ Area   │ Nitrate Density   │\n",
            "├────────┼───────────────────┤\n",
            "│ Area 1 │ 1.00 mg/L         │\n",
            "│ Area 2 │ 1.00 mg/L         │\n",
            "│ Area 3 │ 1.00 mg/L         │\n",
            "╰────────┴───────────────────╯\n",
            "\n",
            "Water Allocation and Usage:\n",
            "╭─────────────┬──────────────────────────────────────────┬─────────────────────────────────────────────────╮\n",
            "│ Agent       │ Allocated Water (million cubic meters)   │ Legal Withdrawal Limit (million cubic meters)   │\n",
            "├─────────────┼──────────────────────────────────────────┼─────────────────────────────────────────────────┤\n",
            "│ farmer_1    │ 100.00 million cubic meters              │ 50.00 million cubic meters                      │\n",
            "│ farmer_2    │ 80.00 million cubic meters               │ 50.00 million cubic meters                      │\n",
            "│ industry_1  │ 380.00 million cubic meters              │ 152.00 million cubic meters                     │\n",
            "│ industry_2  │ 1382.00 million cubic meters             │ 552.80 million cubic meters                     │\n",
            "│ industry_3  │ 238.00 million cubic meters              │ 95.20 million cubic meters                      │\n",
            "│ municipal_1 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_2 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_3 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_4 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_5 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "╰─────────────┴──────────────────────────────────────────┴─────────────────────────────────────────────────╯\n",
            "\n",
            "Green Spaces Managed by Municipal Agents:\n",
            "╭─────────────┬────────────────────────────╮\n",
            "│ Agent       │ Current Green Space Area   │\n",
            "├─────────────┼────────────────────────────┤\n",
            "│ municipal_1 │ 100000.00 square meters    │\n",
            "│ municipal_2 │ 75000.00 square meters     │\n",
            "│ municipal_3 │ 90000.00 square meters     │\n",
            "│ municipal_4 │ 80000.00 square meters     │\n",
            "│ municipal_5 │ 70000.00 square meters     │\n",
            "╰─────────────┴────────────────────────────╯\n",
            "\n",
            "Rewards:\n",
            "╭─────────────┬──────────╮\n",
            "│ Agent       │   Reward │\n",
            "├─────────────┼──────────┤\n",
            "│ industry_1  │        1 │\n",
            "│ industry_2  │        0 │\n",
            "│ industry_3  │        0 │\n",
            "│ municipal_1 │        0 │\n",
            "│ municipal_2 │        0 │\n",
            "│ municipal_3 │        0 │\n",
            "│ municipal_4 │        0 │\n",
            "│ municipal_5 │        0 │\n",
            "│ farmer_1    │        0 │\n",
            "│ farmer_2    │        0 │\n",
            "╰─────────────┴──────────╯\n",
            "----------------------------------------------------------------------------------------------\n",
            "\n",
            "=======================================================\n",
            "                  Testing industry_2\n",
            "=======================================================\n",
            "Initial observation for industry_2: [1169 1058 1001]\n",
            "Action:  (1,)\n",
            "Updated Groundwater levels:  [1168 1044 1001]\n",
            "Observation after action for industry_2: [1168 1044 1001]\n",
            "\n",
            "Current Environment step: 2\n",
            "\n",
            "Groundwater Levels:\n",
            "╭────────┬──────────────────────────────╮\n",
            "│ Area   │ Groundwater Level            │\n",
            "├────────┼──────────────────────────────┤\n",
            "│ Area 1 │ 1168.00 million cubic meters │\n",
            "│ Area 2 │ 1044.00 million cubic meters │\n",
            "│ Area 3 │ 1001.00 million cubic meters │\n",
            "╰────────┴──────────────────────────────╯\n",
            "\n",
            "Nitrate Density:\n",
            "╭────────┬───────────────────╮\n",
            "│ Area   │ Nitrate Density   │\n",
            "├────────┼───────────────────┤\n",
            "│ Area 1 │ 1.00 mg/L         │\n",
            "│ Area 2 │ 1.00 mg/L         │\n",
            "│ Area 3 │ 1.00 mg/L         │\n",
            "╰────────┴───────────────────╯\n",
            "\n",
            "Water Allocation and Usage:\n",
            "╭─────────────┬──────────────────────────────────────────┬─────────────────────────────────────────────────╮\n",
            "│ Agent       │ Allocated Water (million cubic meters)   │ Legal Withdrawal Limit (million cubic meters)   │\n",
            "├─────────────┼──────────────────────────────────────────┼─────────────────────────────────────────────────┤\n",
            "│ farmer_1    │ 100.00 million cubic meters              │ 50.00 million cubic meters                      │\n",
            "│ farmer_2    │ 80.00 million cubic meters               │ 50.00 million cubic meters                      │\n",
            "│ industry_1  │ 380.00 million cubic meters              │ 152.00 million cubic meters                     │\n",
            "│ industry_2  │ 1382.00 million cubic meters             │ 552.80 million cubic meters                     │\n",
            "│ industry_3  │ 238.00 million cubic meters              │ 95.20 million cubic meters                      │\n",
            "│ municipal_1 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_2 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_3 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_4 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_5 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "╰─────────────┴──────────────────────────────────────────┴─────────────────────────────────────────────────╯\n",
            "\n",
            "Green Spaces Managed by Municipal Agents:\n",
            "╭─────────────┬────────────────────────────╮\n",
            "│ Agent       │ Current Green Space Area   │\n",
            "├─────────────┼────────────────────────────┤\n",
            "│ municipal_1 │ 100000.00 square meters    │\n",
            "│ municipal_2 │ 75000.00 square meters     │\n",
            "│ municipal_3 │ 90000.00 square meters     │\n",
            "│ municipal_4 │ 80000.00 square meters     │\n",
            "│ municipal_5 │ 70000.00 square meters     │\n",
            "╰─────────────┴────────────────────────────╯\n",
            "\n",
            "Rewards:\n",
            "╭─────────────┬──────────╮\n",
            "│ Agent       │   Reward │\n",
            "├─────────────┼──────────┤\n",
            "│ industry_1  │        0 │\n",
            "│ industry_2  │        0 │\n",
            "│ industry_3  │       -1 │\n",
            "│ municipal_1 │       -1 │\n",
            "│ municipal_2 │       -1 │\n",
            "│ municipal_3 │       -1 │\n",
            "│ municipal_4 │       -1 │\n",
            "│ municipal_5 │       -1 │\n",
            "│ farmer_1    │       -1 │\n",
            "│ farmer_2    │       -1 │\n",
            "╰─────────────┴──────────╯\n",
            "----------------------------------------------------------------------------------------------\n",
            "\n",
            "=======================================================\n",
            "                  Testing industry_3\n",
            "=======================================================\n",
            "Initial observation for industry_3: [1168 1044 1001]\n",
            "Action:  (1,)\n",
            "Updated Groundwater levels:  [1167 1044  998]\n",
            "Observation after action for industry_3: [1167 1044  998]\n",
            "\n",
            "Current Environment step: 3\n",
            "\n",
            "Groundwater Levels:\n",
            "╭────────┬──────────────────────────────╮\n",
            "│ Area   │ Groundwater Level            │\n",
            "├────────┼──────────────────────────────┤\n",
            "│ Area 1 │ 1167.00 million cubic meters │\n",
            "│ Area 2 │ 1044.00 million cubic meters │\n",
            "│ Area 3 │ 998.00 million cubic meters  │\n",
            "╰────────┴──────────────────────────────╯\n",
            "\n",
            "Nitrate Density:\n",
            "╭────────┬───────────────────╮\n",
            "│ Area   │ Nitrate Density   │\n",
            "├────────┼───────────────────┤\n",
            "│ Area 1 │ 1.00 mg/L         │\n",
            "│ Area 2 │ 1.00 mg/L         │\n",
            "│ Area 3 │ 1.00 mg/L         │\n",
            "╰────────┴───────────────────╯\n",
            "\n",
            "Water Allocation and Usage:\n",
            "╭─────────────┬──────────────────────────────────────────┬─────────────────────────────────────────────────╮\n",
            "│ Agent       │ Allocated Water (million cubic meters)   │ Legal Withdrawal Limit (million cubic meters)   │\n",
            "├─────────────┼──────────────────────────────────────────┼─────────────────────────────────────────────────┤\n",
            "│ farmer_1    │ 100.00 million cubic meters              │ 50.00 million cubic meters                      │\n",
            "│ farmer_2    │ 80.00 million cubic meters               │ 50.00 million cubic meters                      │\n",
            "│ industry_1  │ 380.00 million cubic meters              │ 152.00 million cubic meters                     │\n",
            "│ industry_2  │ 1382.00 million cubic meters             │ 552.80 million cubic meters                     │\n",
            "│ industry_3  │ 238.00 million cubic meters              │ 95.20 million cubic meters                      │\n",
            "│ municipal_1 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_2 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_3 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_4 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_5 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "╰─────────────┴──────────────────────────────────────────┴─────────────────────────────────────────────────╯\n",
            "\n",
            "Green Spaces Managed by Municipal Agents:\n",
            "╭─────────────┬────────────────────────────╮\n",
            "│ Agent       │ Current Green Space Area   │\n",
            "├─────────────┼────────────────────────────┤\n",
            "│ municipal_1 │ 100000.00 square meters    │\n",
            "│ municipal_2 │ 75000.00 square meters     │\n",
            "│ municipal_3 │ 90000.00 square meters     │\n",
            "│ municipal_4 │ 80000.00 square meters     │\n",
            "│ municipal_5 │ 70000.00 square meters     │\n",
            "╰─────────────┴────────────────────────────╯\n",
            "\n",
            "Rewards:\n",
            "╭─────────────┬──────────╮\n",
            "│ Agent       │   Reward │\n",
            "├─────────────┼──────────┤\n",
            "│ industry_1  │       -1 │\n",
            "│ industry_2  │       -1 │\n",
            "│ industry_3  │        0 │\n",
            "│ municipal_1 │       -2 │\n",
            "│ municipal_2 │       -2 │\n",
            "│ municipal_3 │       -2 │\n",
            "│ municipal_4 │       -2 │\n",
            "│ municipal_5 │       -2 │\n",
            "│ farmer_1    │       -2 │\n",
            "│ farmer_2    │       -2 │\n",
            "╰─────────────┴──────────╯\n",
            "----------------------------------------------------------------------------------------------\n",
            "\n",
            "=======================================================\n",
            "                  Testing municipal_1\n",
            "=======================================================\n",
            "Initial observation for municipal_1: [1167 1044  998]\n",
            "Action:  (15, 0, 5)\n",
            "Updated Groundwater levels:  [1166 1044  998]\n",
            "Observation after action for municipal_1: [1166 1044  998]\n",
            "\n",
            "Current Environment step: 4\n",
            "\n",
            "Groundwater Levels:\n",
            "╭────────┬──────────────────────────────╮\n",
            "│ Area   │ Groundwater Level            │\n",
            "├────────┼──────────────────────────────┤\n",
            "│ Area 1 │ 1166.00 million cubic meters │\n",
            "│ Area 2 │ 1044.00 million cubic meters │\n",
            "│ Area 3 │ 998.00 million cubic meters  │\n",
            "╰────────┴──────────────────────────────╯\n",
            "\n",
            "Nitrate Density:\n",
            "╭────────┬───────────────────╮\n",
            "│ Area   │ Nitrate Density   │\n",
            "├────────┼───────────────────┤\n",
            "│ Area 1 │ 1.00 mg/L         │\n",
            "│ Area 2 │ 1.00 mg/L         │\n",
            "│ Area 3 │ 1.00 mg/L         │\n",
            "╰────────┴───────────────────╯\n",
            "\n",
            "Water Allocation and Usage:\n",
            "╭─────────────┬──────────────────────────────────────────┬─────────────────────────────────────────────────╮\n",
            "│ Agent       │ Allocated Water (million cubic meters)   │ Legal Withdrawal Limit (million cubic meters)   │\n",
            "├─────────────┼──────────────────────────────────────────┼─────────────────────────────────────────────────┤\n",
            "│ farmer_1    │ 100.00 million cubic meters              │ 50.00 million cubic meters                      │\n",
            "│ farmer_2    │ 80.00 million cubic meters               │ 50.00 million cubic meters                      │\n",
            "│ industry_1  │ 380.00 million cubic meters              │ 152.00 million cubic meters                     │\n",
            "│ industry_2  │ 1382.00 million cubic meters             │ 552.80 million cubic meters                     │\n",
            "│ industry_3  │ 238.00 million cubic meters              │ 95.20 million cubic meters                      │\n",
            "│ municipal_1 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_2 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_3 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_4 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_5 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "╰─────────────┴──────────────────────────────────────────┴─────────────────────────────────────────────────╯\n",
            "\n",
            "Green Spaces Managed by Municipal Agents:\n",
            "╭─────────────┬────────────────────────────╮\n",
            "│ Agent       │ Current Green Space Area   │\n",
            "├─────────────┼────────────────────────────┤\n",
            "│ municipal_1 │ 120000.00 square meters    │\n",
            "│ municipal_2 │ 75000.00 square meters     │\n",
            "│ municipal_3 │ 90000.00 square meters     │\n",
            "│ municipal_4 │ 80000.00 square meters     │\n",
            "│ municipal_5 │ 70000.00 square meters     │\n",
            "╰─────────────┴────────────────────────────╯\n",
            "\n",
            "Rewards:\n",
            "╭─────────────┬──────────╮\n",
            "│ Agent       │   Reward │\n",
            "├─────────────┼──────────┤\n",
            "│ industry_1  │    -2    │\n",
            "│ industry_2  │    -2    │\n",
            "│ industry_3  │    -1    │\n",
            "│ municipal_1 │    -0.97 │\n",
            "│ municipal_2 │    -3    │\n",
            "│ municipal_3 │    -3    │\n",
            "│ municipal_4 │    -3    │\n",
            "│ municipal_5 │    -3    │\n",
            "│ farmer_1    │    -3    │\n",
            "│ farmer_2    │    -3    │\n",
            "╰─────────────┴──────────╯\n",
            "----------------------------------------------------------------------------------------------\n",
            "\n",
            "=======================================================\n",
            "                  Testing municipal_2\n",
            "=======================================================\n",
            "Initial observation for municipal_2: [1166 1044  998]\n",
            "Action:  (15, 0, 5)\n",
            "Updated Groundwater levels:  [1165 1044  998]\n",
            "Observation after action for municipal_2: [1165 1044  998]\n",
            "\n",
            "Current Environment step: 5\n",
            "\n",
            "Groundwater Levels:\n",
            "╭────────┬──────────────────────────────╮\n",
            "│ Area   │ Groundwater Level            │\n",
            "├────────┼──────────────────────────────┤\n",
            "│ Area 1 │ 1165.00 million cubic meters │\n",
            "│ Area 2 │ 1044.00 million cubic meters │\n",
            "│ Area 3 │ 998.00 million cubic meters  │\n",
            "╰────────┴──────────────────────────────╯\n",
            "\n",
            "Nitrate Density:\n",
            "╭────────┬───────────────────╮\n",
            "│ Area   │ Nitrate Density   │\n",
            "├────────┼───────────────────┤\n",
            "│ Area 1 │ 1.00 mg/L         │\n",
            "│ Area 2 │ 1.00 mg/L         │\n",
            "│ Area 3 │ 1.00 mg/L         │\n",
            "╰────────┴───────────────────╯\n",
            "\n",
            "Water Allocation and Usage:\n",
            "╭─────────────┬──────────────────────────────────────────┬─────────────────────────────────────────────────╮\n",
            "│ Agent       │ Allocated Water (million cubic meters)   │ Legal Withdrawal Limit (million cubic meters)   │\n",
            "├─────────────┼──────────────────────────────────────────┼─────────────────────────────────────────────────┤\n",
            "│ farmer_1    │ 100.00 million cubic meters              │ 50.00 million cubic meters                      │\n",
            "│ farmer_2    │ 80.00 million cubic meters               │ 50.00 million cubic meters                      │\n",
            "│ industry_1  │ 380.00 million cubic meters              │ 152.00 million cubic meters                     │\n",
            "│ industry_2  │ 1382.00 million cubic meters             │ 552.80 million cubic meters                     │\n",
            "│ industry_3  │ 238.00 million cubic meters              │ 95.20 million cubic meters                      │\n",
            "│ municipal_1 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_2 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_3 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_4 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_5 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "╰─────────────┴──────────────────────────────────────────┴─────────────────────────────────────────────────╯\n",
            "\n",
            "Green Spaces Managed by Municipal Agents:\n",
            "╭─────────────┬────────────────────────────╮\n",
            "│ Agent       │ Current Green Space Area   │\n",
            "├─────────────┼────────────────────────────┤\n",
            "│ municipal_1 │ 120000.00 square meters    │\n",
            "│ municipal_2 │ 90000.00 square meters     │\n",
            "│ municipal_3 │ 90000.00 square meters     │\n",
            "│ municipal_4 │ 80000.00 square meters     │\n",
            "│ municipal_5 │ 70000.00 square meters     │\n",
            "╰─────────────┴────────────────────────────╯\n",
            "\n",
            "Rewards:\n",
            "╭─────────────┬──────────╮\n",
            "│ Agent       │   Reward │\n",
            "├─────────────┼──────────┤\n",
            "│ industry_1  │    -3    │\n",
            "│ industry_2  │    -3    │\n",
            "│ industry_3  │    -2    │\n",
            "│ municipal_1 │    -1.97 │\n",
            "│ municipal_2 │    -0.96 │\n",
            "│ municipal_3 │    -4    │\n",
            "│ municipal_4 │    -4    │\n",
            "│ municipal_5 │    -4    │\n",
            "│ farmer_1    │    -4    │\n",
            "│ farmer_2    │    -4    │\n",
            "╰─────────────┴──────────╯\n",
            "----------------------------------------------------------------------------------------------\n",
            "\n",
            "=======================================================\n",
            "                  Testing municipal_3\n",
            "=======================================================\n",
            "Initial observation for municipal_3: [1165 1044  998]\n",
            "Action:  (15, 0, 5)\n",
            "Updated Groundwater levels:  [1164 1044  998]\n",
            "Observation after action for municipal_3: [1164 1044  998]\n",
            "\n",
            "Current Environment step: 6\n",
            "\n",
            "Groundwater Levels:\n",
            "╭────────┬──────────────────────────────╮\n",
            "│ Area   │ Groundwater Level            │\n",
            "├────────┼──────────────────────────────┤\n",
            "│ Area 1 │ 1164.00 million cubic meters │\n",
            "│ Area 2 │ 1044.00 million cubic meters │\n",
            "│ Area 3 │ 998.00 million cubic meters  │\n",
            "╰────────┴──────────────────────────────╯\n",
            "\n",
            "Nitrate Density:\n",
            "╭────────┬───────────────────╮\n",
            "│ Area   │ Nitrate Density   │\n",
            "├────────┼───────────────────┤\n",
            "│ Area 1 │ 1.00 mg/L         │\n",
            "│ Area 2 │ 1.00 mg/L         │\n",
            "│ Area 3 │ 1.00 mg/L         │\n",
            "╰────────┴───────────────────╯\n",
            "\n",
            "Water Allocation and Usage:\n",
            "╭─────────────┬──────────────────────────────────────────┬─────────────────────────────────────────────────╮\n",
            "│ Agent       │ Allocated Water (million cubic meters)   │ Legal Withdrawal Limit (million cubic meters)   │\n",
            "├─────────────┼──────────────────────────────────────────┼─────────────────────────────────────────────────┤\n",
            "│ farmer_1    │ 100.00 million cubic meters              │ 50.00 million cubic meters                      │\n",
            "│ farmer_2    │ 80.00 million cubic meters               │ 50.00 million cubic meters                      │\n",
            "│ industry_1  │ 380.00 million cubic meters              │ 152.00 million cubic meters                     │\n",
            "│ industry_2  │ 1382.00 million cubic meters             │ 552.80 million cubic meters                     │\n",
            "│ industry_3  │ 238.00 million cubic meters              │ 95.20 million cubic meters                      │\n",
            "│ municipal_1 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_2 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_3 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_4 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_5 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "╰─────────────┴──────────────────────────────────────────┴─────────────────────────────────────────────────╯\n",
            "\n",
            "Green Spaces Managed by Municipal Agents:\n",
            "╭─────────────┬────────────────────────────╮\n",
            "│ Agent       │ Current Green Space Area   │\n",
            "├─────────────┼────────────────────────────┤\n",
            "│ municipal_1 │ 120000.00 square meters    │\n",
            "│ municipal_2 │ 90000.00 square meters     │\n",
            "│ municipal_3 │ 108000.00 square meters    │\n",
            "│ municipal_4 │ 80000.00 square meters     │\n",
            "│ municipal_5 │ 70000.00 square meters     │\n",
            "╰─────────────┴────────────────────────────╯\n",
            "\n",
            "Rewards:\n",
            "╭─────────────┬──────────╮\n",
            "│ Agent       │   Reward │\n",
            "├─────────────┼──────────┤\n",
            "│ industry_1  │    -4    │\n",
            "│ industry_2  │    -4    │\n",
            "│ industry_3  │    -3    │\n",
            "│ municipal_1 │    -2.97 │\n",
            "│ municipal_2 │    -1.96 │\n",
            "│ municipal_3 │    -0.97 │\n",
            "│ municipal_4 │    -5    │\n",
            "│ municipal_5 │    -5    │\n",
            "│ farmer_1    │    -5    │\n",
            "│ farmer_2    │    -5    │\n",
            "╰─────────────┴──────────╯\n",
            "----------------------------------------------------------------------------------------------\n",
            "\n",
            "=======================================================\n",
            "                  Testing municipal_4\n",
            "=======================================================\n",
            "Initial observation for municipal_4: [1164 1044  998]\n",
            "Action:  (15, 0, 5)\n",
            "Updated Groundwater levels:  [1163 1044  998]\n",
            "Observation after action for municipal_4: [1163 1044  998]\n",
            "\n",
            "Current Environment step: 7\n",
            "\n",
            "Groundwater Levels:\n",
            "╭────────┬──────────────────────────────╮\n",
            "│ Area   │ Groundwater Level            │\n",
            "├────────┼──────────────────────────────┤\n",
            "│ Area 1 │ 1163.00 million cubic meters │\n",
            "│ Area 2 │ 1044.00 million cubic meters │\n",
            "│ Area 3 │ 998.00 million cubic meters  │\n",
            "╰────────┴──────────────────────────────╯\n",
            "\n",
            "Nitrate Density:\n",
            "╭────────┬───────────────────╮\n",
            "│ Area   │ Nitrate Density   │\n",
            "├────────┼───────────────────┤\n",
            "│ Area 1 │ 1.00 mg/L         │\n",
            "│ Area 2 │ 1.00 mg/L         │\n",
            "│ Area 3 │ 1.00 mg/L         │\n",
            "╰────────┴───────────────────╯\n",
            "\n",
            "Water Allocation and Usage:\n",
            "╭─────────────┬──────────────────────────────────────────┬─────────────────────────────────────────────────╮\n",
            "│ Agent       │ Allocated Water (million cubic meters)   │ Legal Withdrawal Limit (million cubic meters)   │\n",
            "├─────────────┼──────────────────────────────────────────┼─────────────────────────────────────────────────┤\n",
            "│ farmer_1    │ 100.00 million cubic meters              │ 50.00 million cubic meters                      │\n",
            "│ farmer_2    │ 80.00 million cubic meters               │ 50.00 million cubic meters                      │\n",
            "│ industry_1  │ 380.00 million cubic meters              │ 152.00 million cubic meters                     │\n",
            "│ industry_2  │ 1382.00 million cubic meters             │ 552.80 million cubic meters                     │\n",
            "│ industry_3  │ 238.00 million cubic meters              │ 95.20 million cubic meters                      │\n",
            "│ municipal_1 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_2 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_3 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_4 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_5 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "╰─────────────┴──────────────────────────────────────────┴─────────────────────────────────────────────────╯\n",
            "\n",
            "Green Spaces Managed by Municipal Agents:\n",
            "╭─────────────┬────────────────────────────╮\n",
            "│ Agent       │ Current Green Space Area   │\n",
            "├─────────────┼────────────────────────────┤\n",
            "│ municipal_1 │ 120000.00 square meters    │\n",
            "│ municipal_2 │ 90000.00 square meters     │\n",
            "│ municipal_3 │ 108000.00 square meters    │\n",
            "│ municipal_4 │ 96000.00 square meters     │\n",
            "│ municipal_5 │ 70000.00 square meters     │\n",
            "╰─────────────┴────────────────────────────╯\n",
            "\n",
            "Rewards:\n",
            "╭─────────────┬──────────╮\n",
            "│ Agent       │   Reward │\n",
            "├─────────────┼──────────┤\n",
            "│ industry_1  │    -5    │\n",
            "│ industry_2  │    -5    │\n",
            "│ industry_3  │    -4    │\n",
            "│ municipal_1 │    -3.97 │\n",
            "│ municipal_2 │    -2.96 │\n",
            "│ municipal_3 │    -1.97 │\n",
            "│ municipal_4 │    -0.96 │\n",
            "│ municipal_5 │    -6    │\n",
            "│ farmer_1    │    -6    │\n",
            "│ farmer_2    │    -6    │\n",
            "╰─────────────┴──────────╯\n",
            "----------------------------------------------------------------------------------------------\n",
            "\n",
            "=======================================================\n",
            "                  Testing municipal_5\n",
            "=======================================================\n",
            "Initial observation for municipal_5: [1163 1044  998]\n",
            "Action:  (15, 0, 5)\n",
            "Updated Groundwater levels:  [1162 1044  998]\n",
            "Observation after action for municipal_5: [1162 1044  998]\n",
            "\n",
            "Current Environment step: 8\n",
            "\n",
            "Groundwater Levels:\n",
            "╭────────┬──────────────────────────────╮\n",
            "│ Area   │ Groundwater Level            │\n",
            "├────────┼──────────────────────────────┤\n",
            "│ Area 1 │ 1162.00 million cubic meters │\n",
            "│ Area 2 │ 1044.00 million cubic meters │\n",
            "│ Area 3 │ 998.00 million cubic meters  │\n",
            "╰────────┴──────────────────────────────╯\n",
            "\n",
            "Nitrate Density:\n",
            "╭────────┬───────────────────╮\n",
            "│ Area   │ Nitrate Density   │\n",
            "├────────┼───────────────────┤\n",
            "│ Area 1 │ 1.00 mg/L         │\n",
            "│ Area 2 │ 1.00 mg/L         │\n",
            "│ Area 3 │ 1.00 mg/L         │\n",
            "╰────────┴───────────────────╯\n",
            "\n",
            "Water Allocation and Usage:\n",
            "╭─────────────┬──────────────────────────────────────────┬─────────────────────────────────────────────────╮\n",
            "│ Agent       │ Allocated Water (million cubic meters)   │ Legal Withdrawal Limit (million cubic meters)   │\n",
            "├─────────────┼──────────────────────────────────────────┼─────────────────────────────────────────────────┤\n",
            "│ farmer_1    │ 100.00 million cubic meters              │ 50.00 million cubic meters                      │\n",
            "│ farmer_2    │ 80.00 million cubic meters               │ 50.00 million cubic meters                      │\n",
            "│ industry_1  │ 380.00 million cubic meters              │ 152.00 million cubic meters                     │\n",
            "│ industry_2  │ 1382.00 million cubic meters             │ 552.80 million cubic meters                     │\n",
            "│ industry_3  │ 238.00 million cubic meters              │ 95.20 million cubic meters                      │\n",
            "│ municipal_1 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_2 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_3 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_4 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_5 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "╰─────────────┴──────────────────────────────────────────┴─────────────────────────────────────────────────╯\n",
            "\n",
            "Green Spaces Managed by Municipal Agents:\n",
            "╭─────────────┬────────────────────────────╮\n",
            "│ Agent       │ Current Green Space Area   │\n",
            "├─────────────┼────────────────────────────┤\n",
            "│ municipal_1 │ 120000.00 square meters    │\n",
            "│ municipal_2 │ 90000.00 square meters     │\n",
            "│ municipal_3 │ 108000.00 square meters    │\n",
            "│ municipal_4 │ 96000.00 square meters     │\n",
            "│ municipal_5 │ 84000.00 square meters     │\n",
            "╰─────────────┴────────────────────────────╯\n",
            "\n",
            "Rewards:\n",
            "╭─────────────┬──────────╮\n",
            "│ Agent       │   Reward │\n",
            "├─────────────┼──────────┤\n",
            "│ industry_1  │    -6    │\n",
            "│ industry_2  │    -6    │\n",
            "│ industry_3  │    -5    │\n",
            "│ municipal_1 │    -4.97 │\n",
            "│ municipal_2 │    -3.96 │\n",
            "│ municipal_3 │    -2.97 │\n",
            "│ municipal_4 │    -1.96 │\n",
            "│ municipal_5 │    -0.96 │\n",
            "│ farmer_1    │    -7    │\n",
            "│ farmer_2    │    -7    │\n",
            "╰─────────────┴──────────╯\n",
            "----------------------------------------------------------------------------------------------\n",
            "\n",
            "=======================================================\n",
            "                  Testing farmer_1\n",
            "=======================================================\n",
            "Initial observation for farmer_1: [1162 1044  998]\n",
            "Action:  ([50, 30, 20], 25, 0)\n",
            "Updated Groundwater levels:  [1161  894  998]\n",
            "Observation after action for farmer_1: [1161  894  998]\n",
            "\n",
            "Current Environment step: 9\n",
            "\n",
            "Groundwater Levels:\n",
            "╭────────┬──────────────────────────────╮\n",
            "│ Area   │ Groundwater Level            │\n",
            "├────────┼──────────────────────────────┤\n",
            "│ Area 1 │ 1161.00 million cubic meters │\n",
            "│ Area 2 │ 894.00 million cubic meters  │\n",
            "│ Area 3 │ 998.00 million cubic meters  │\n",
            "╰────────┴──────────────────────────────╯\n",
            "\n",
            "Nitrate Density:\n",
            "╭────────┬───────────────────╮\n",
            "│ Area   │ Nitrate Density   │\n",
            "├────────┼───────────────────┤\n",
            "│ Area 1 │ 1.00 mg/L         │\n",
            "│ Area 2 │ 1.00 mg/L         │\n",
            "│ Area 3 │ 1.00 mg/L         │\n",
            "╰────────┴───────────────────╯\n",
            "\n",
            "Water Allocation and Usage:\n",
            "╭─────────────┬──────────────────────────────────────────┬─────────────────────────────────────────────────╮\n",
            "│ Agent       │ Allocated Water (million cubic meters)   │ Legal Withdrawal Limit (million cubic meters)   │\n",
            "├─────────────┼──────────────────────────────────────────┼─────────────────────────────────────────────────┤\n",
            "│ farmer_1    │ 100.00 million cubic meters              │ 50.00 million cubic meters                      │\n",
            "│ farmer_2    │ 80.00 million cubic meters               │ 50.00 million cubic meters                      │\n",
            "│ industry_1  │ 380.00 million cubic meters              │ 152.00 million cubic meters                     │\n",
            "│ industry_2  │ 1382.00 million cubic meters             │ 552.80 million cubic meters                     │\n",
            "│ industry_3  │ 238.00 million cubic meters              │ 95.20 million cubic meters                      │\n",
            "│ municipal_1 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_2 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_3 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_4 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_5 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "╰─────────────┴──────────────────────────────────────────┴─────────────────────────────────────────────────╯\n",
            "\n",
            "Green Spaces Managed by Municipal Agents:\n",
            "╭─────────────┬────────────────────────────╮\n",
            "│ Agent       │ Current Green Space Area   │\n",
            "├─────────────┼────────────────────────────┤\n",
            "│ municipal_1 │ 120000.00 square meters    │\n",
            "│ municipal_2 │ 90000.00 square meters     │\n",
            "│ municipal_3 │ 108000.00 square meters    │\n",
            "│ municipal_4 │ 96000.00 square meters     │\n",
            "│ municipal_5 │ 84000.00 square meters     │\n",
            "╰─────────────┴────────────────────────────╯\n",
            "\n",
            "Rewards:\n",
            "╭─────────────┬──────────╮\n",
            "│ Agent       │   Reward │\n",
            "├─────────────┼──────────┤\n",
            "│ industry_1  │    -7    │\n",
            "│ industry_2  │    -7    │\n",
            "│ industry_3  │    -6    │\n",
            "│ municipal_1 │    -5.97 │\n",
            "│ municipal_2 │    -4.96 │\n",
            "│ municipal_3 │    -3.97 │\n",
            "│ municipal_4 │    -2.96 │\n",
            "│ municipal_5 │    -1.96 │\n",
            "│ farmer_1    │    -0.96 │\n",
            "│ farmer_2    │    -8    │\n",
            "╰─────────────┴──────────╯\n",
            "----------------------------------------------------------------------------------------------\n",
            "\n",
            "=======================================================\n",
            "                  Testing farmer_2\n",
            "=======================================================\n",
            "Initial observation for farmer_2: [1161  894  998]\n",
            "Action:  ([50, 30, 20], 25, 0)\n",
            "Updated Groundwater levels:  [1160  894  868]\n",
            "Observation after action for farmer_2: [1160  894  868]\n",
            "\n",
            "Current Environment step: 10\n",
            "\n",
            "Groundwater Levels:\n",
            "╭────────┬──────────────────────────────╮\n",
            "│ Area   │ Groundwater Level            │\n",
            "├────────┼──────────────────────────────┤\n",
            "│ Area 1 │ 1160.00 million cubic meters │\n",
            "│ Area 2 │ 894.00 million cubic meters  │\n",
            "│ Area 3 │ 868.00 million cubic meters  │\n",
            "╰────────┴──────────────────────────────╯\n",
            "\n",
            "Nitrate Density:\n",
            "╭────────┬───────────────────╮\n",
            "│ Area   │ Nitrate Density   │\n",
            "├────────┼───────────────────┤\n",
            "│ Area 1 │ 1.00 mg/L         │\n",
            "│ Area 2 │ 1.00 mg/L         │\n",
            "│ Area 3 │ 1.00 mg/L         │\n",
            "╰────────┴───────────────────╯\n",
            "\n",
            "Water Allocation and Usage:\n",
            "╭─────────────┬──────────────────────────────────────────┬─────────────────────────────────────────────────╮\n",
            "│ Agent       │ Allocated Water (million cubic meters)   │ Legal Withdrawal Limit (million cubic meters)   │\n",
            "├─────────────┼──────────────────────────────────────────┼─────────────────────────────────────────────────┤\n",
            "│ farmer_1    │ 100.00 million cubic meters              │ 50.00 million cubic meters                      │\n",
            "│ farmer_2    │ 80.00 million cubic meters               │ 50.00 million cubic meters                      │\n",
            "│ industry_1  │ 380.00 million cubic meters              │ 152.00 million cubic meters                     │\n",
            "│ industry_2  │ 1382.00 million cubic meters             │ 552.80 million cubic meters                     │\n",
            "│ industry_3  │ 238.00 million cubic meters              │ 95.20 million cubic meters                      │\n",
            "│ municipal_1 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_2 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_3 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_4 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "│ municipal_5 │ 200.00 million cubic meters              │ 80.00 million cubic meters                      │\n",
            "╰─────────────┴──────────────────────────────────────────┴─────────────────────────────────────────────────╯\n",
            "\n",
            "Green Spaces Managed by Municipal Agents:\n",
            "╭─────────────┬────────────────────────────╮\n",
            "│ Agent       │ Current Green Space Area   │\n",
            "├─────────────┼────────────────────────────┤\n",
            "│ municipal_1 │ 120000.00 square meters    │\n",
            "│ municipal_2 │ 90000.00 square meters     │\n",
            "│ municipal_3 │ 108000.00 square meters    │\n",
            "│ municipal_4 │ 96000.00 square meters     │\n",
            "│ municipal_5 │ 84000.00 square meters     │\n",
            "╰─────────────┴────────────────────────────╯\n",
            "\n",
            "Rewards:\n",
            "╭─────────────┬──────────╮\n",
            "│ Agent       │   Reward │\n",
            "├─────────────┼──────────┤\n",
            "│ industry_1  │    -8    │\n",
            "│ industry_2  │    -8    │\n",
            "│ industry_3  │    -7    │\n",
            "│ municipal_1 │    -6.97 │\n",
            "│ municipal_2 │    -5.96 │\n",
            "│ municipal_3 │    -4.97 │\n",
            "│ municipal_4 │    -3.96 │\n",
            "│ municipal_5 │    -2.96 │\n",
            "│ farmer_1    │    -1.96 │\n",
            "│ farmer_2    │    -0.96 │\n",
            "╰─────────────┴──────────╯\n",
            "----------------------------------------------------------------------------------------------\n",
            "\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [],
      "metadata": {
        "id": "i0lyvwIi6ku9"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}