diff --git a/scripts/run-example-notebook-test.sh b/scripts/run-example-notebook-test.sh new file mode 100755 index 000000000..56970d597 --- /dev/null +++ b/scripts/run-example-notebook-test.sh @@ -0,0 +1,3 @@ +#!/usr/bin/env bash + +pytest --junitxml /tmp/notebook-test-results.xml --log-cli-level INFO --nbmake examples/notebooks diff --git a/scripts/run-learning-test.sh b/scripts/run-learning-test.sh new file mode 100755 index 000000000..6ca85183c --- /dev/null +++ b/scripts/run-learning-test.sh @@ -0,0 +1,3 @@ +#!/usr/bin/env bash + +python -m vega_sim.reinforcement.run_rl_agent --rl-max-it $1 diff --git a/vega_sim/null_service.py b/vega_sim/null_service.py index 2dbbc984b..e8a9e5223 100644 --- a/vega_sim/null_service.py +++ b/vega_sim/null_service.py @@ -385,6 +385,25 @@ def manage_vega_processes( data_node_docker_volume = docker_client.volumes.create() data_node_container = docker_client.containers.run( "timescale/timescaledb:2.8.0-pg14", + [ + "postgres", + "-c", + "max_connections=50", + "-c", + "log_destination=stderr", + "-c", + "work_mem=5MB", + "-c", + "huge_pages=off", + "-c", + "shared_memory_type=sysv", + "-c", + "dynamic_shared_memory_type=sysv", + "-c", + "shared_buffers=2GB", + "-c", + "temp_buffers=5MB", + ], detach=True, ports={5432: port_config[Ports.DATA_NODE_POSTGRES]}, volumes=[f"{data_node_docker_volume.name}:/var/lib/postgresql/data"],