Gymnasium/bin/docker_entrypoint

#!/bin/sh
# This script is the entrypoint for our Docker image.

set -e

# Inject the mujoco license key (if planning to publish image, this step has to be runtime)
echo $MUJOCO_KEY | base64 --decode > /root/.mujoco/mjkey.txt

# Set up display; otherwise rendering will fail
Xvfb -screen 0 1024x768x24 &
export DISPLAY=:0

# Wait for the file to come up
display=0
file="/tmp/.X11-unix/X$display"
for i in $(seq 1 10); do
    if [ -e "$file" ]; then
	break
    fi

    echo "Waiting for $file to be created (try $i/10)"
    sleep "$i"
done
if ! [ -e "$file" ]; then
    echo "Timing out: $file was not created"
    exit 1
fi

exec "$@"