mirror of
https://github.com/gsi-upm/soil
synced 2025-07-11 07:22:21 +00:00
9749f4ca14
Multithreading needs pickling to work. Pickling/unpickling didn't work in some situations, like when the environment_agents parameter was left blank. This was due to two reasons: 1) agents and history didn't have a setstate method, and some of their attributes cannot be pickled (generators, sqlite connection) 2) the environment was adding generators (agents) to its state. This fixes the situation by restricting the keys that the environment exports when it pickles, and by adding the set/getstate methods in agents. The resulting pickles should contain enough information to inspect them (history, state values, etc), but very limited.
77 lines
2.5 KiB
Python
77 lines
2.5 KiB
Python
import importlib
|
|
import sys
|
|
import os
|
|
import pdb
|
|
import logging
|
|
|
|
from .version import __version__
|
|
|
|
try:
|
|
basestring
|
|
except NameError:
|
|
basestring = str
|
|
|
|
from . import agents
|
|
from .simulation import *
|
|
from .environment import Environment
|
|
from . import utils
|
|
from . import analysis
|
|
|
|
def main():
|
|
import argparse
|
|
from . import simulation
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logging.info('Running SOIL version: {}'.format(__version__))
|
|
|
|
parser = argparse.ArgumentParser(description='Run a SOIL simulation')
|
|
parser.add_argument('file', type=str,
|
|
nargs="?",
|
|
default='simulation.yml',
|
|
help='python module containing the simulation configuration.')
|
|
parser.add_argument('--module', '-m', type=str,
|
|
help='file containing the code of any custom agents.')
|
|
parser.add_argument('--dry-run', '--dry', action='store_true',
|
|
help='Do not store the results of the simulation.')
|
|
parser.add_argument('--pdb', action='store_true',
|
|
help='Use a pdb console in case of exception.')
|
|
parser.add_argument('--graph', '-g', action='store_true',
|
|
help='Dump GEXF graph. Defaults to false.')
|
|
parser.add_argument('--csv', action='store_true',
|
|
help='Dump history in CSV format. Defaults to false.')
|
|
parser.add_argument('--output', '-o', type=str, default="soil_output",
|
|
help='folder to write results to. It defaults to the current directory.')
|
|
parser.add_argument('--synchronous', action='store_true',
|
|
help='Run trials serially and synchronously instead of in parallel. Defaults to false.')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if os.getcwd() not in sys.path:
|
|
sys.path.append(os.getcwd())
|
|
if args.module:
|
|
importlib.import_module(args.module)
|
|
|
|
logging.info('Loading config file: {}'.format(args.file))
|
|
|
|
try:
|
|
dump = []
|
|
if not args.dry_run:
|
|
if args.csv:
|
|
dump.append('csv')
|
|
if args.graph:
|
|
dump.append('gexf')
|
|
simulation.run_from_config(args.file,
|
|
dry_run=args.dry_run,
|
|
dump=dump,
|
|
parallel=(not args.synchronous),
|
|
results_dir=args.output)
|
|
except Exception:
|
|
if args.pdb:
|
|
pdb.post_mortem()
|
|
else:
|
|
raise
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|