The following shouldn't materialize all previous blocks:

ray.data.range(1000, parallelism=1000).map(lambda x: x).limit(1).materialize()

This can be implemented with a special Limit operator that terminates once it receives enough elements.