Allowing nd inputs

Allowing n-dimensional inputs to the PySR algorithm by passing a boolean corresponding to the allow_nd boolean in sklearn's validation methods.