SafePolicyImprovementNonstationary/run_python.sh at main · Bulbatronik/SafePolicyImprovementNonstationary · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#!/bin/bash

# If, during docker run command, you encounter errors related to the shell script syntax,
# use this "dos2unix file_name";

# Get the current platform OS (Linux, macOS, Windows, Docker) as a user parameter
PLATFORM_OS=$1
# Get the number of specified processes as a user parameter
N_PROC=$2
ID_LIST=$3

if [ "$PLATFORM_OS" = "Docker" ]; then
    # Set the python executable path
    PYTHON_EXECUTABLE="/usr/bin/python3"

    # Run the Python script
    python3 ./run.py \
        --init \
        --pythonexec $PYTHON_EXECUTABLE \
        --n-proc $N_PROC \
        -f bandit_swarm.jl \
        -o log_dir \
        -s 0 \
        --speeds 0,1,2,3 \
        --ids $ID_LIST \
        --trials 10 \
        --eps 100
        # As stated in the paper, the values are:
        # eps: 2000
        # ids: 1000
        # speeds: 0,1,2,3
        # trials: 10
else
    # Get the current directory absolute path
    CURRENT_DIR=$(pwd)
    # Name of the virtual environment
    ENV_NAME=".venv"
    # Check if the environment folder already exists
    if [ ! -d "$ENV_NAME" ]; then
        # The environment doesn't exist, create it
        python -m venv $ENV_NAME
    fi
    # Check if the platform OS is Linux or macOS
    if [ "$PLATFORM_OS" = "Linux" ] || [ "$PLATFORM_OS" = "macOS" ]; then
        # Set the python executable path
        PYTHON_EXECUTABLE="$CURRENT_DIR/$ENV_NAME/bin/python"
        # Activate the environment
        source $ENV_NAME/bin/activate
    fi
    if [ "$PLATFORM_OS" = "Windows" ]; then
        # Set the python executable path
        PYTHON_EXECUTABLE="$CURRENT_DIR/$ENV_NAME/Scripts/python.exe"
        # Activate the environment
        source $ENV_NAME/Scripts/activate
    fi

    INVENV=$(python -c 'import sys; print ("1" if sys.prefix != sys.base_prefix else "0")')
    # If the INVENV is set to 0 (not activated)
    if [ "$INVENV" == "0" ]; then
        echo "The environment is not activated"
        exit 1
    fi

    # Run the Python script
    python ./run.py \
        --init \
        --pythonexec $PYTHON_EXECUTABLE \
        --n-proc $N_PROC \
        -f bandit_swarm.jl \
        -o log_dir \
        -s 0 \
        --speeds 0,1,2,3 \
        --ids $ID_LIST \
        --trials 10 \
        --eps 2000
        # As stated in the paper, the values are:
        # eps: 2000
        # ids: 1000
        # speeds: 0,1,2,3
        # trials: 10
fi

# Check if the platform OS is Linux or macOS or Windows
if [ "$PLATFORM_OS" = "Linux" ] || [ "$PLATFORM_OS" = "macOS" ]; then
    # Deactivate the environment
    deactivate
fi