Change infer_schema API to allow querying for a single stream

Update this docstring?