ambr 0.1.0__tar.gz

ambr-0.1.0/LICENSE

@@ -0,0 +1,29 @@
+BSD 3-Clause License
+
+Copyright (c) 2026, AMBER Team
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

ambr-0.1.0/MANIFEST.in

@@ -0,0 +1,6 @@
+include README.md
+include LICENSE
+
+include architecture_diagram.png
+include requirements.txt
+recursive-include src/amber *.py

ambr-0.1.0/PKG-INFO

@@ -0,0 +1,202 @@
+Metadata-Version: 2.4
+Name: ambr
+Version: 0.1.0
+Summary: Agent-Based Modeling Environment for Research
+Home-page: https://github.com/amber-team/amber
+Author: AMBER Team
+Author-email: amber@example.com
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: polars>=0.20.0
+Requires-Dist: numpy>=1.20.0
+Requires-Dist: matplotlib>=3.3.0
+Requires-Dist: seaborn>=0.11.0
+Requires-Dist: networkx>=2.5
+Requires-Dist: scikit-optimize>=0.9.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: pytest-mock>=3.10.0; extra == "dev"
+Requires-Dist: coverage>=7.0.0; extra == "dev"
+Requires-Dist: sphinx>=4.0.0; extra == "dev"
+Requires-Dist: sphinx-rtd-theme>=1.0.0; extra == "dev"
+Requires-Dist: myst-parser>=0.18.0; extra == "dev"
+Requires-Dist: nbsphinx>=0.8.0; extra == "dev"
+Requires-Dist: nbconvert>=6.0.0; extra == "dev"
+Requires-Dist: jupyter>=1.0.0; extra == "dev"
+Requires-Dist: black>=22.0.0; extra == "dev"
+Requires-Dist: flake8>=5.0.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Provides-Extra: advanced
+Requires-Dist: smac>=2.0.0; extra == "advanced"
+Requires-Dist: ConfigSpace>=0.7.1; extra == "advanced"
+Requires-Dist: scikit-learn>=1.0.0; extra == "advanced"
+Provides-Extra: docs
+Requires-Dist: sphinx>=4.0.0; extra == "docs"
+Requires-Dist: sphinx-rtd-theme>=1.0.0; extra == "docs"
+Requires-Dist: myst-parser>=0.18.0; extra == "docs"
+Requires-Dist: nbsphinx>=0.8.0; extra == "docs"
+Requires-Dist: nbconvert>=6.0.0; extra == "docs"
+Requires-Dist: jupyter>=1.0.0; extra == "docs"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# AMBER (Agent-based Modeling with Blazingly Efficient Records)
+
+[![CI](https://github.com/a11to1n3/AMBER/actions/workflows/ci.yml/badge.svg)](https://github.com/a11to1n3/AMBER/actions/workflows/ci.yml)
+[![codecov](https://codecov.io/gh/a11to1n3/AMBER/graph/badge.svg)](https://codecov.io/gh/a11to1n3/AMBER)
+[![Python 3.9+](https://img.shields.io/badge/python-3.9+-blue.svg)](https://www.python.org/downloads/release/python-390/)
+[![License: BSD-3-Clause](https://img.shields.io/badge/License-BSD_3_Clause-blue.svg)](https://opensource.org/licenses/BSD-3-Clause)
+
+AMBER is a Python framework for agent-based modeling that uses Polars for efficient data handling and analysis. AMBER provides a clean, robust API for creating parallel, high-performance simulations in Python.
+
+## 🚀 Performance
+
+AMBER achieves high performance through its columnar memory layout (SoA), KD-Trees, and SIMD-vectorized operations.
+
+**Benchmark vs Other Frameworks (5,000 Agents, 100 Steps)**
+
+| Framework | Language | Architecture | Speed Rank |
+|-----------|----------|--------------|------------|
+| **Agents.jl** | Julia | Vectorized | 🥇 1st (0.02s - 0.9s) |
+| **AMBER** | Python | Vectorized | 🥈 **2nd (0.1s - 2.3s)** |
+| **SimPy (Dense)** | Python | Process/DES | 🥉 3rd (0.3s - 4.3s) |
+| **Melodie** | Python | Hybrid | 4th (0.4s - 20s) |
+| **AgentPy** | Python | Object | 5th (2s - 30s) |
+| **Mesa** | Python | Object | 6th (50s - 30s) |
+
+
+*\*SimPy is exceptionally fast for sparse models (like SIR) due to its event-driven nature, but AMBER is faster for dense/movement-heavy models.*
+
+AMBER is the **state-of-the-art for dense simulation** in Python, while SimPy offers an alternative for event-driven logic.
+
+![Comparison Chart](benchmarks/results/scaling_chart.png)
+
+## 🚀 Quick Start
+
+```python
+import amber as am
+
+# Define a custom agent
+class MyAgent(am.Agent):
+    def setup(self):
+        self.value = self.p.initial_value
+        
+    def step(self):
+        self.value += self.model.random.randint(-1, 2)
+        self.record('value', self.value)
+
+# Define a model
+class MyModel(am.Model):
+    def setup(self):
+        self.agents = am.AgentList(self, self.p.agents, MyAgent)
+        
+    def step(self):
+        self.agents.call('step')
+        
+    def update(self):
+        super().update()
+        self.agents.record('value')
+
+# Run a simulation
+parameters = {
+    'agents': 10,
+    'initial_value': 5,
+    'steps': 100
+}
+
+model = MyModel(parameters)
+results = model.run()
+```
+
+## ⚡ Advanced: Vectorized Updates
+
+For maximum performance, access the underlying `Population` manager to perform SIMD-vectorized state updates, bypassing Python loop overhead:
+
+```python
+def step(self):
+    # Create a batch update context
+    with self.population.create_batch_context() as batch:
+        # Queue updates logic (executed in Rust/Polars)
+        batch.add_update(target_ids, 'wealth', 1)
+        batch.add_update(source_ids, 'wealth', -1)
+    
+    # State is applied atomically here
+```
+
+## 🔬 Optimization
+
+AMBER includes powerful optimization capabilities for parameter tuning:
+
+```python
+from amber.optimization import ParameterSpace, grid_search
+
+# Define parameter space
+parameter_space = ParameterSpace({
+    'agents': [10, 50, 100],
+    'initial_value': [1, 5, 10],
+    'steps': 100
+})
+
+# Run optimization
+results = grid_search(MyModel, parameter_space, 'some_metric')
+best_params = results[0]['parameters']
+```
+
+## 📦 Installation
+
+```bash
+pip install -e .
+```
+
+## 🏗️ Features
+
+- **Simple API**: Intuitive interface for agent-based modeling
+- **High Performance**: Efficient data handling with Polars DataFrames
+- **Optimization**: Built-in parameter optimization with grid search, random search, and Bayesian optimization
+- **Environments**: Support for grid, network, and continuous space environments
+- **Experiments**: Run multiple simulations with parameter sampling
+- **Random Number Generation**: Reproducible simulations with controlled randomness
+
+## 📚 Examples
+
+Working examples are available in the `examples/` directory:
+
+- **Wealth Transfer Model**: Economic inequality simulation
+- **Virus Spread Model**: Epidemiological SIR model
+- **Flocking Simulation**: Boids flocking behavior
+- **Forest Fire Model**: Cellular automata fire spread
+- **Network Simulations**: Graph-based agent interactions
+
+## 📖 Documentation
+
+For detailed documentation, tutorials, and API reference, visit our documentation site.
+
+## 🤝 Contributing
+
+We welcome contributions! Please see our contributing guidelines for more information.
+
+## 📄 License
+
+This project is licensed under the BSD 3-Clause License - see the [LICENSE](LICENSE) file for details.

ambr-0.1.0/README.md

@@ -0,0 +1,138 @@
+# AMBER (Agent-based Modeling with Blazingly Efficient Records)
+
+[![CI](https://github.com/a11to1n3/AMBER/actions/workflows/ci.yml/badge.svg)](https://github.com/a11to1n3/AMBER/actions/workflows/ci.yml)
+[![codecov](https://codecov.io/gh/a11to1n3/AMBER/graph/badge.svg)](https://codecov.io/gh/a11to1n3/AMBER)
+[![Python 3.9+](https://img.shields.io/badge/python-3.9+-blue.svg)](https://www.python.org/downloads/release/python-390/)
+[![License: BSD-3-Clause](https://img.shields.io/badge/License-BSD_3_Clause-blue.svg)](https://opensource.org/licenses/BSD-3-Clause)
+
+AMBER is a Python framework for agent-based modeling that uses Polars for efficient data handling and analysis. AMBER provides a clean, robust API for creating parallel, high-performance simulations in Python.
+
+## 🚀 Performance
+
+AMBER achieves high performance through its columnar memory layout (SoA), KD-Trees, and SIMD-vectorized operations.
+
+**Benchmark vs Other Frameworks (5,000 Agents, 100 Steps)**
+
+| Framework | Language | Architecture | Speed Rank |
+|-----------|----------|--------------|------------|
+| **Agents.jl** | Julia | Vectorized | 🥇 1st (0.02s - 0.9s) |
+| **AMBER** | Python | Vectorized | 🥈 **2nd (0.1s - 2.3s)** |
+| **SimPy (Dense)** | Python | Process/DES | 🥉 3rd (0.3s - 4.3s) |
+| **Melodie** | Python | Hybrid | 4th (0.4s - 20s) |
+| **AgentPy** | Python | Object | 5th (2s - 30s) |
+| **Mesa** | Python | Object | 6th (50s - 30s) |
+
+
+*\*SimPy is exceptionally fast for sparse models (like SIR) due to its event-driven nature, but AMBER is faster for dense/movement-heavy models.*
+
+AMBER is the **state-of-the-art for dense simulation** in Python, while SimPy offers an alternative for event-driven logic.
+
+![Comparison Chart](benchmarks/results/scaling_chart.png)
+
+## 🚀 Quick Start
+
+```python
+import amber as am
+
+# Define a custom agent
+class MyAgent(am.Agent):
+    def setup(self):
+        self.value = self.p.initial_value
+        
+    def step(self):
+        self.value += self.model.random.randint(-1, 2)
+        self.record('value', self.value)
+
+# Define a model
+class MyModel(am.Model):
+    def setup(self):
+        self.agents = am.AgentList(self, self.p.agents, MyAgent)
+        
+    def step(self):
+        self.agents.call('step')
+        
+    def update(self):
+        super().update()
+        self.agents.record('value')
+
+# Run a simulation
+parameters = {
+    'agents': 10,
+    'initial_value': 5,
+    'steps': 100
+}
+
+model = MyModel(parameters)
+results = model.run()
+```
+
+## ⚡ Advanced: Vectorized Updates
+
+For maximum performance, access the underlying `Population` manager to perform SIMD-vectorized state updates, bypassing Python loop overhead:
+
+```python
+def step(self):
+    # Create a batch update context
+    with self.population.create_batch_context() as batch:
+        # Queue updates logic (executed in Rust/Polars)
+        batch.add_update(target_ids, 'wealth', 1)
+        batch.add_update(source_ids, 'wealth', -1)
+    
+    # State is applied atomically here
+```
+
+## 🔬 Optimization
+
+AMBER includes powerful optimization capabilities for parameter tuning:
+
+```python
+from amber.optimization import ParameterSpace, grid_search
+
+# Define parameter space
+parameter_space = ParameterSpace({
+    'agents': [10, 50, 100],
+    'initial_value': [1, 5, 10],
+    'steps': 100
+})
+
+# Run optimization
+results = grid_search(MyModel, parameter_space, 'some_metric')
+best_params = results[0]['parameters']
+```
+
+## 📦 Installation
+
+```bash
+pip install -e .
+```
+
+## 🏗️ Features
+
+- **Simple API**: Intuitive interface for agent-based modeling
+- **High Performance**: Efficient data handling with Polars DataFrames
+- **Optimization**: Built-in parameter optimization with grid search, random search, and Bayesian optimization
+- **Environments**: Support for grid, network, and continuous space environments
+- **Experiments**: Run multiple simulations with parameter sampling
+- **Random Number Generation**: Reproducible simulations with controlled randomness
+
+## 📚 Examples
+
+Working examples are available in the `examples/` directory:
+
+- **Wealth Transfer Model**: Economic inequality simulation
+- **Virus Spread Model**: Epidemiological SIR model
+- **Flocking Simulation**: Boids flocking behavior
+- **Forest Fire Model**: Cellular automata fire spread
+- **Network Simulations**: Graph-based agent interactions
+
+## 📖 Documentation
+
+For detailed documentation, tutorials, and API reference, visit our documentation site.
+
+## 🤝 Contributing
+
+We welcome contributions! Please see our contributing guidelines for more information.
+
+## 📄 License
+
+This project is licensed under the BSD 3-Clause License - see the [LICENSE](LICENSE) file for details.

ambr-0.1.0/requirements.txt

@@ -0,0 +1,21 @@
+# Core dependencies
+polars>=0.20.0
+numpy>=1.20.0
+networkx>=2.5
+
+# Visualization
+matplotlib>=3.3.0
+seaborn>=0.11.0
+
+# Optional optimization dependencies
+scikit-optimize>=0.9.0
+
+# Development and documentation dependencies (optional)
+# Uncomment for development:
+# pytest>=6.0.0
+# pytest-cov>=2.0.0
+# sphinx>=4.0.0
+# sphinx-rtd-theme>=1.0.0
+# myst-parser>=0.18.0
+# nbsphinx>=0.8.0
+# jupyter>=1.0.0 

ambr-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

ambr-0.1.0/setup.py

@@ -0,0 +1,66 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="ambr",
+    version="0.1.0",
+    packages=find_packages(where="src"),
+    package_dir={"": "src"},
+    install_requires=[
+        "polars>=0.20.0",
+        "numpy>=1.20.0",
+        "matplotlib>=3.3.0",
+        "seaborn>=0.11.0",
+        "networkx>=2.5",
+        "scikit-optimize>=0.9.0",
+    ],
+    extras_require={
+        "dev": [
+            "pytest>=7.0.0",
+            "pytest-cov>=4.0.0",
+            "pytest-mock>=3.10.0",
+            "coverage>=7.0.0",
+            "sphinx>=4.0.0",
+            "sphinx-rtd-theme>=1.0.0",
+            "myst-parser>=0.18.0",
+            "nbsphinx>=0.8.0",
+            "nbconvert>=6.0.0",
+            "jupyter>=1.0.0",
+            "black>=22.0.0",
+            "flake8>=5.0.0",
+            "mypy>=1.0.0",
+        ],
+        "advanced": [
+            "smac>=2.0.0",
+            "ConfigSpace>=0.7.1",
+            "scikit-learn>=1.0.0",
+        ],
+        "docs": [
+            "sphinx>=4.0.0",
+            "sphinx-rtd-theme>=1.0.0",
+            "myst-parser>=0.18.0",
+            "nbsphinx>=0.8.0",
+            "nbconvert>=6.0.0",
+            "jupyter>=1.0.0",
+        ],
+    },
+    python_requires=">=3.8",
+    author="AMBER Team",
+    author_email="amber@example.com",
+    description="Agent-Based Modeling Environment for Research",
+    long_description=open("README.md").read(),
+    long_description_content_type="text/markdown",
+    url="https://github.com/amber-team/amber",
+    classifiers=[
+        "Development Status :: 4 - Beta",
+        "Intended Audience :: Science/Research",
+        "License :: OSI Approved :: BSD License",
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
+        "Programming Language :: Python :: 3.12",
+        "Topic :: Scientific/Engineering",
+        "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    ],
+) 

ambr-0.1.0/src/amber/__init__.py

@@ -0,0 +1,105 @@
+"""
+AMBER: Agent-Based Modeling Environment and Research Framework
+
+A comprehensive Python framework for building, running, and analyzing agent-based models.
+AMBER provides tools for creating complex simulations with agents, environments, and 
+sophisticated experimental workflows.
+
+Key features:
+- Flexible agent and model architecture
+- Built-in environments (grid, space, network)
+- Experiment management and parameter sweeping
+- Optimization tools for model calibration
+- High-performance data handling with Polars
+
+Example:
+    >>> import amber as am
+    >>> 
+    >>> class SimpleModel(am.Model):
+    ...     def setup(self):
+    ...         for i in range(10):
+    ...             agent = am.Agent(self, i)
+    ...             self.add_agent(agent)
+    ...     
+    ...     def step(self):
+    ...         self.record_model('agent_count', len(self.agents))
+    >>> 
+    >>> model = SimpleModel({'steps': 5})
+    >>> results = model.run()
+"""
+
+from .agent import Agent
+from .model import Model
+from .population import Population, BatchUpdateContext
+from .base import BaseModel, BaseAgent
+from .sequences import AgentList
+from .environments import GridEnvironment, SpaceEnvironment, NetworkEnvironment
+from .experiment import Experiment, Sample, IntRange
+from .optimization import (
+    ParameterSpace, 
+    objective_function,
+    grid_search,
+    random_search,
+    bayesian_optimization,
+    SMACOptimizer,
+    MultiObjectiveSMAC,
+    SMACParameterSpace
+)
+from .performance import (
+    SpatialIndex,
+    ParallelRunner,
+    vectorized_move,
+    vectorized_wealth_transfer,
+    vectorized_random_velocities,
+    check_performance_deps,
+    HAS_SCIPY,
+    HAS_NUMBA,
+)
+
+__version__ = '0.1.0'
+__author__ = 'AMBER Team'
+__email__ = 'amber@example.com'
+__url__ = 'https://github.com/example/amber'
+
+__all__ = [
+    'Agent',
+    'Model',
+    'Population',
+    'BatchUpdateContext',
+    'BaseModel',
+    'BaseAgent',
+    'AgentList',
+    'GridEnvironment',
+    'SpaceEnvironment', 
+    'NetworkEnvironment',
+    'Experiment',
+    'Sample',
+    'IntRange',
+    'ParameterSpace',
+    'objective_function',
+    'grid_search',
+    'random_search',
+    'bayesian_optimization',
+    'SMACOptimizer',
+    'MultiObjectiveSMAC',
+    'SMACParameterSpace',
+    # Performance utilities
+    'SpatialIndex',
+    'ParallelRunner',
+    'vectorized_move',
+    'vectorized_wealth_transfer',
+    'vectorized_random_velocities',
+    'check_performance_deps',
+    'HAS_SCIPY',
+    'HAS_NUMBA',
+]
+
+# Clean up namespace - remove module references
+import sys
+_current_module = sys.modules[__name__]
+for _name in list(globals().keys()):
+    if _name.startswith('_'):
+        continue
+    if _name not in __all__ and _name not in ['__version__', '__author__', '__email__', '__url__']:
+        delattr(_current_module, _name)
+del _current_module, _name 

ambr-0.1.0/src/amber/agent.py

@@ -0,0 +1,65 @@
+from typing import Any, Dict, Optional
+import polars as pl
+from .base import BaseAgent
+
+class Agent(BaseAgent):
+    """Base class for all agents in the simulation, using DataFrames for data storage."""
+    
+    def __repr__(self):
+        """String representation of the agent."""
+        return f"Agent(id={self.id})"
+    
+    def setup(self):
+        """Initialize agent attributes. Override this method in subclasses."""
+        pass
+        
+    def record(self, name: str, value: Any):
+        """Record a variable value for this agent in the model's DataFrame.
+        
+        Args:
+            name: Name of the variable
+            value: Value to record
+        """
+        # Add new column to model's DataFrame
+        self.model.agents_df = self.model.agents_df.with_columns([
+            pl.when(pl.col('id') == self.id)
+            .then(pl.lit(value))
+            .otherwise(pl.col(name))
+            .alias(name)
+        ])
+        
+    def get_data(self) -> pl.DataFrame:
+        """Get this agent's data from the model's DataFrame.
+        
+        Returns:
+            DataFrame containing this agent's data
+        """
+        return self.model.agents_df.filter(pl.col('id') == self.id)
+        
+    def update_data(self, data: Dict[str, Any]):
+        """Update this agent's data in the model's DataFrame.
+        
+        Args:
+            data: Dictionary of column names and values to update
+        """
+        for name, value in data.items():
+            self.model.agents_df = self.model.agents_df.with_columns([
+                pl.when(pl.col('id') == self.id)
+                .then(pl.lit(value))
+                .otherwise(pl.col(name))
+                .alias(name)
+            ])
+            
+    def get_neighbors(self, condition: Optional[pl.Expr] = None) -> pl.DataFrame:
+        """Get neighboring agents' data based on a condition.
+        
+        Args:
+            condition: Optional Polars expression to filter neighbors
+            
+        Returns:
+            DataFrame containing neighboring agents' data
+        """
+        neighbors = self.model.agents_df.filter(pl.col('id') != self.id)
+        if condition is not None:
+            neighbors = neighbors.filter(condition)
+        return neighbors 

ambr-0.1.0/src/amber/base.py

@@ -0,0 +1,24 @@
+from typing import Any, Dict, Optional, List, Type
+import polars as pl
+import numpy as np
+import random
+
+class BaseModel:
+    """Base class for all simulation models, using DataFrames for data storage."""
+    def __init__(self, parameters: Dict[str, Any]):
+        self.p = parameters
+        self.t = 0
+        self.agents_df = pl.DataFrame({'id': [], 't': []})
+        self.model_df = pl.DataFrame({'t': [0], **{k: [v] for k, v in parameters.items()}})
+        seed = parameters.get('seed', None)
+        self.random = random.Random(seed)
+        self.nprandom = np.random.default_rng(seed)
+
+class BaseAgent:
+    """Base class for all agents in the simulation, using DataFrames for data storage."""
+    def __init__(self, model: 'BaseModel', agent_id: int):
+        self.model = model
+        self.id = agent_id
+        self.p = model.p if model is not None else {}
+
+