Merge branch 'master' into bump/deepspeed

Author: Borda

.actions/assistant.py

@@ -341,19 +341,6 @@ def create_mirror_package(source_dir: str, package_mapping: dict[str, str]) -> N
 
 
 class AssistantCLI:
-    @staticmethod
-    def _replace_min(fname: str) -> None:
-        with open(fname, encoding="utf-8") as fopen:
-            req = fopen.read().replace(">=", "==")
-        with open(fname, "w", encoding="utf-8") as fwrite:
-            fwrite.write(req)
-
-    @staticmethod
-    def replace_oldest_ver(requirement_fnames: Sequence[str] = REQUIREMENT_FILES_ALL) -> None:
-        """Replace the min package version by fixed one."""
-        for fname in requirement_fnames:
-            print(fname)
-            AssistantCLI._replace_min(fname)
 
     @staticmethod
     def copy_replace_imports(

.azure/gpu-tests-fabric.yml

@@ -99,7 +99,9 @@ jobs:
         displayName: "Image info & NVIDIA"
 
       - bash: |
-          python .actions/assistant.py replace_oldest_ver
+          cd requirements/fabric
+          pip install -U "lightning-utilities[cli]"
+          python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'strategies.txt']"
           pip install "cython<3.0" wheel  # for compatibility
           pip install -U "lightning-utilities[cli]"
           # drop deepspeed since it is not supported by our minimal Torch requirements

.azure/gpu-tests-pytorch.yml

@@ -103,7 +103,9 @@ jobs:
         displayName: "Image info & NVIDIA"
 
       - bash: |
-          python .actions/assistant.py replace_oldest_ver
+          cd requirements/pytorch
+          pip install -U "lightning-utilities[cli]"
+          python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'extra.txt', 'strategies.txt', 'examples.txt']"
           pip install "cython<3.0" wheel  # for compatibility
           pip install -U "lightning-utilities[cli]"
           # drop deepspeed since it is not supported by our minimal Torch requirements

.github/workflows/ci-tests-fabric.yml

@@ -94,7 +94,9 @@ jobs:
       - name: Set min. dependencies
         if: ${{ matrix.requires == 'oldest' }}
         run: |
-          python .actions/assistant.py replace_oldest_ver
+          cd requirements/fabric
+          pip install -U "lightning-utilities[cli]"
+          python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'strategies.txt']"
           pip install "cython<3.0" wheel
           pip install "pyyaml==5.4" --no-build-isolation
 

.github/workflows/ci-tests-pytorch.yml

@@ -99,7 +99,9 @@ jobs:
       - name: Set min. dependencies
         if: ${{ matrix.requires == 'oldest' }}
         run: |
-          python .actions/assistant.py replace_oldest_ver
+          cd requirements/pytorch
+          pip install -U "lightning-utilities[cli]"
+          python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'extra.txt', 'strategies.txt', 'examples.txt']"
           pip install "cython<3.0" wheel
           pip install "pyyaml==5.4" --no-build-isolation
 

.lightning/workflows/fabric.yml

@@ -43,7 +43,10 @@ run: |
 
   if [ "${TORCH_VER}" == "2.1" ]; then
     echo "Set oldest versions"
-    python .actions/assistant.py replace_oldest_ver
+    cd requirements/fabric
+    pip install -U "lightning-utilities[cli]"
+    python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'strategies.txt']"
+    cd ../..
     pip install "cython<3.0" wheel  # for compatibility
   fi
 

.lightning/workflows/pytorch.yml

@@ -43,7 +43,10 @@ run: |
 
   if [ "${TORCH_VER}" == "2.1" ]; then
     recho "Set oldest versions"
-    python .actions/assistant.py replace_oldest_ver
+    cd requirements/pytorch
+    pip install -U "lightning-utilities[cli]"
+    python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'extra.txt', 'strategies.txt', 'examples.txt']"
+    cd ../..
     pip install "cython<3.0" wheel  # for compatibility
   fi