Does not work with --partition alias

[shreyasgm]

Thank you for this wonderful script. Just wanted to highlight a minor issue with this script.

The command find-best-partition -f run_jupyter.sh -o check:

Works when the header for run_jupyter.sh looks like this:

#!/bin/bash
#SBATCH -p test
...

But does not work when run_jupyter.sh starts like this:

#!/bin/bash
#SBATCH --partition=test
...

You would expect the code to run fine in both cases.

[mmshad]

Thank you @shreyasgm and sorry for checking the issue late. I'll fix this in the next version.