Check extra vars before running inference

Currently if you define some extra variables to write out at inference time that don't exist in the test file, the inference is run first and this only crashes at the writing step. Instead we should check this during the setup.