diff --git a/litdata/__about__.py b/litdata/__about__.py
index e1c19f1d..b2dd5728 100644
--- a/litdata/__about__.py
+++ b/litdata/__about__.py
@@ -14,7 +14,7 @@
 
 import time
 
-__version__ = "0.2.0.rc"
+__version__ = "0.2.0.rc1"
 __author__ = "Lightning AI et al."
 __author_email__ = "pytorch@lightning.ai"
 __license__ = "Apache-2.0"
diff --git a/requirements.txt b/requirements.txt
index 40a47c42..5d286375 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,8 +1,6 @@
-lightning-utilities >=0.8.0, <0.10.0
 lightning-cloud == 0.5.64  # Must be pinned to ensure compatibility
-# to be able to include also PL 2.0 and preserve `>` needed for CI min version bypass
+lightning-utilities >=0.8.0, <0.10.0
 torch >=2.1.0, <=2.2.0
-lightning >=2.2.0
 filelock
 tqdm
 numpy
diff --git a/requirements/extras.txt b/requirements/extras.txt
new file mode 100644
index 00000000..e69de29b
diff --git a/requirements/test.txt b/requirements/test.txt
index 58e5caad..c3e97950 100644
--- a/requirements/test.txt
+++ b/requirements/test.txt
@@ -5,3 +5,4 @@ pytest-timeout ==2.1.0
 pytest-rerunfailures ==12.0
 pytest-random-order ==1.1.0
 pandas
+lightning