Fix the CI pipeline with the new container

Some python packages, like fireworks, have been removed from the container.