Can we upgrade the datasets lib version? It is no longer supported in other libraries like trl, which requires datasets >= 3.0.0