lightning 2.2.0.post0__tar.gz → 2.2.1__tar.gz

{lightning-2.2.0.post0/src/lightning.egg-info → lightning-2.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning
-Version: 2.2.0.post0
+Version: 2.2.1
 Summary: The Deep Learning framework to train, deploy, and ship AI products Lightning fast.
 Home-page: https://github.com/Lightning-AI/lightning
 Author: Lightning AI et al.
@@ -27,21 +27,21 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Provides-Extra: data-cloud
+Provides-Extra: app-test
+Provides-Extra: app-components
+Provides-Extra: app-cloud
+Provides-Extra: app-ui
 Provides-Extra: data-test
+Provides-Extra: data-cloud
 Provides-Extra: data-examples
+Provides-Extra: fabric-test
+Provides-Extra: fabric-strategies
+Provides-Extra: fabric-examples
+Provides-Extra: store-test
 Provides-Extra: pytorch-test
-Provides-Extra: pytorch-examples
 Provides-Extra: pytorch-strategies
 Provides-Extra: pytorch-extra
-Provides-Extra: store-test
-Provides-Extra: app-cloud
-Provides-Extra: app-test
-Provides-Extra: app-ui
-Provides-Extra: app-components
-Provides-Extra: fabric-test
-Provides-Extra: fabric-examples
-Provides-Extra: fabric-strategies
+Provides-Extra: pytorch-examples
 Provides-Extra: fabric-all
 Provides-Extra: fabric-dev
 Provides-Extra: pytorch-all
@@ -51,15 +51,15 @@ Provides-Extra: app-all
 Provides-Extra: app-dev
 Provides-Extra: data-all
 Provides-Extra: data-dev
-Provides-Extra: cloud
 Provides-Extra: test
+Provides-Extra: components
+Provides-Extra: app
+Provides-Extra: cloud
+Provides-Extra: ui
 Provides-Extra: data
 Provides-Extra: examples
 Provides-Extra: strategies
 Provides-Extra: extra
-Provides-Extra: app
-Provides-Extra: ui
-Provides-Extra: components
 Provides-Extra: all
 Provides-Extra: dev
 Provides-Extra: store
@@ -83,7 +83,7 @@ ______________________________________________________________________
   <a href="https://lightning.ai/docs/pytorch/stable/">PyTorch Lightning</a> •
   <a href="https://lightning.ai/docs/fabric/stable/">Fabric</a> •
   <a href="https://lightning.ai/docs/app/stable/">Lightning Apps</a> •
-  <a href="https://pytorch-lightning.readthedocs.io/en/2.2.0.post0">Docs</a> •
+  <a href="https://pytorch-lightning.readthedocs.io/en/2.2.1">Docs</a> •
   <a href="#community">Community</a> •
   <a href="https://lightning.ai/docs/pytorch/stable/generated/CONTRIBUTING.html">Contribute</a> •
 </p>
@@ -626,7 +626,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
 
 |       System / PyTorch ver.        | 1.13                                                                                                                                                                                                                            | 2.0                                                                                                                                                                                                                             |                                                                                                               2.1                                                                                                               |
 | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
-|        Linux py3.9 \[GPUs\]        |  |  | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.2.0.post0)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
+|        Linux py3.9 \[GPUs\]        |  |  | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.2.1)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
 |        Linux py3.9 \[TPUs\]        |                                                                                                                                                                                                                                 |  [![Test PyTorch - TPU](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml)     |      |
 |  Linux (multiple Python versions)  | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 |
 |   OSX (multiple Python versions)   | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 |

{lightning-2.2.0.post0 → lightning-2.2.1}/setup.py

@@ -40,6 +40,7 @@ There are considered three main scenarios for installing this project:
     c) validate packages and publish to PyPI
 
 """
+
 import contextlib
 import glob
 import logging

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/__init__.py

@@ -1,4 +1,5 @@
 """Root package info."""
+
 import logging
 import sys
 

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/__init__.py

@@ -1,4 +1,5 @@
 """Root package info."""
+
 import logging
 import os
 

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/cli/app-template/tests/test_placeholdername_app.py

@@ -6,6 +6,7 @@ To test a lightning app:
 
 run_once runs your app through one cycle of the event loop and then terminates
 """
+
 import io
 import os
 from contextlib import redirect_stdout

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/cli/component-template/tests/test_placeholdername_component.py

@@ -4,6 +4,7 @@ r"""To test a lightning component:
 2. call .run()
 
 """
+
 from placeholdername.component import TemplateComponent
 
 

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/cli/pl-app-template/core/callbacks.py

@@ -297,9 +297,11 @@ class PLAppArtifactsTracker(Callback):
         for logger in trainer.loggers:
             metadata = {"class_name": logger.__class__.__name__}
             if isinstance(logger, WandbLogger) and not logger._offline:
-                metadata.update(
-                    {"username": logger.experiment.entity, "project_name": logger.name, "run_id": logger.version}
-                )
+                metadata.update({
+                    "username": logger.experiment.entity,
+                    "project_name": logger.name,
+                    "run_id": logger.version,
+                })
 
             if metadata and metadata not in self.work.logger_metadatas:
                 self.work.logger_metadatas.append(metadata)

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/cli/pl-app-template/core/components/logger/tensorboard.py

@@ -32,17 +32,15 @@ class TensorBoardWorker(LightningWork):
         self._sync_every_n_seconds = sync_every_n_seconds
 
     def run(self) -> None:
-        subprocess.Popen(
-            [
-                "tensorboard",
-                "--logdir",
-                str(self.log_dir),
-                "--host",
-                self.host,
-                "--port",
-                str(self.port),
-            ]
-        )
+        subprocess.Popen([
+            "tensorboard",
+            "--logdir",
+            str(self.log_dir),
+            "--host",
+            self.host,
+            "--port",
+            str(self.port),
+        ])
 
         # Download the log directory periodically
         while True:

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/cli/pl-app-template/core/components/script_runner/script_runner.py

@@ -37,14 +37,12 @@ class ScriptRunner(TracerPythonScript):
 
         def pre_trainer_init(_, *args: Any, **kwargs: Any) -> Tuple[Dict, Tuple[Any, ...], Dict[str, Any]]:
             kwargs.setdefault("callbacks", [])
-            kwargs["callbacks"].extend(
-                [
-                    trainer_artifacts_tracker,
-                    trainer_state_tracker,
-                    progress_tracker,
-                    summary,
-                ]
-            )
+            kwargs["callbacks"].extend([
+                trainer_artifacts_tracker,
+                trainer_state_tracker,
+                progress_tracker,
+                summary,
+            ])
             return {}, args, kwargs
 
         tracer.add_traced(Trainer, "__init__", pre_fn=pre_trainer_init)

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/components/database/utilities.py

@@ -148,23 +148,19 @@ class _GeneralModel(BaseModel):
 
     @classmethod
     def from_obj(cls, obj, token):
-        return cls(
-            **{
-                "cls_name": obj.__class__.__name__,
-                "data": obj.json(),
-                "token": token,
-            }
-        )
+        return cls(**{
+            "cls_name": obj.__class__.__name__,
+            "data": obj.json(),
+            "token": token,
+        })
 
     @classmethod
     def from_cls(cls, obj_cls, token):
-        return cls(
-            **{
-                "cls_name": obj_cls.__name__,
-                "data": "",
-                "token": token,
-            }
-        )
+        return cls(**{
+            "cls_name": obj_cls.__name__,
+            "data": "",
+            "token": token,
+        })
 
 
 class _SelectAll:

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/components/multi_node/base.py

@@ -77,17 +77,15 @@ class MultiNode(LightningFlow):
                 " To run on multiple nodes in the cloud, launch your app with `--cloud`."
             )
             num_nodes = 1
-        self.ws = _List(
-            *[
-                work_cls(
-                    *work_args,
-                    cloud_compute=cloud_compute.clone(),
-                    **work_kwargs,
-                    parallel=True,
-                )
-                for _ in range(num_nodes)
-            ]
-        )
+        self.ws = _List(*[
+            work_cls(
+                *work_args,
+                cloud_compute=cloud_compute.clone(),
+                **work_kwargs,
+                parallel=True,
+            )
+            for _ in range(num_nodes)
+        ])
 
     def run(self) -> None:
         # 1. Wait for all works to be started !

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/components/multi_node/fabric.py

@@ -28,8 +28,7 @@ from lightning.app.utilities.tracer import Tracer
 @runtime_checkable
 class _FabricWorkProtocol(Protocol):
     @staticmethod
-    def run() -> None:
-        ...
+    def run() -> None: ...
 
 
 @dataclass

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/components/multi_node/trainer.py

@@ -28,8 +28,7 @@ from lightning.app.utilities.tracer import Tracer
 @runtime_checkable
 class _LightningTrainerWorkProtocol(Protocol):
     @staticmethod
-    def run() -> None:
-        ...
+    def run() -> None: ...
 
 
 @dataclass

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/components/serve/auto_scaler.py

@@ -614,12 +614,10 @@ class AutoScaler(LightningFlow):
     def create_work(self) -> LightningWork:
         """Replicates a LightningWork instance with args and kwargs provided via ``__init__``."""
         cloud_compute = self._work_kwargs.get("cloud_compute", None)
-        self._work_kwargs.update(
-            {
-                "start_with_flow": False,
-                "cloud_compute": cloud_compute.clone() if cloud_compute else None,
-            }
-        )
+        self._work_kwargs.update({
+            "start_with_flow": False,
+            "cloud_compute": cloud_compute.clone() if cloud_compute else None,
+        })
         return self._work_cls(*self._work_args, **self._work_kwargs)
 
     def add_work(self, work) -> str:

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/core/app.py

@@ -350,10 +350,10 @@ class LightningApp:
 
         while (time() - t0) < self.state_accumulate_wait:
             # TODO: Fetch all available deltas at once to reduce queue calls.
-            received_deltas: List[
-                Union[_DeltaRequest, _APIRequest, _CommandRequest, ComponentDelta]
-            ] = self.batch_get_state_changed_from_queue(
-                self.delta_queue  # type: ignore[assignment,arg-type]
+            received_deltas: List[Union[_DeltaRequest, _APIRequest, _CommandRequest, ComponentDelta]] = (
+                self.batch_get_state_changed_from_queue(
+                    self.delta_queue  # type: ignore[assignment,arg-type]
+                )
             )
             if len(received_deltas) == []:
                 break
@@ -371,7 +371,9 @@ class LightningApp:
 
                     if work:
                         delta = _delta_to_app_state_delta(
-                            self.root, work, deepcopy(delta.delta)  # type: ignore[arg-type]
+                            self.root,  # type: ignore[arg-type]
+                            work,
+                            deepcopy(delta.delta),
                         )
                         deltas.append(delta)
                 else:

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/frontend/panel/__init__.py

@@ -1,4 +1,5 @@
 """The PanelFrontend and AppStateWatcher make it easy to create Lightning Apps with the Panel data app framework."""
+
 from lightning.app.frontend.panel.app_state_watcher import AppStateWatcher
 from lightning.app.frontend.panel.panel_frontend import PanelFrontend
 

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/frontend/panel/app_state_comm.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """The watch_app_state function enables us to trigger a callback function when ever the app state changes."""
+
 # Todo: Refactor with Streamlit
 # Note: It would be nice one day to just watch changes within the Flow scope instead of whole app
 from __future__ import annotations

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/frontend/panel/app_state_watcher.py

@@ -19,6 +19,7 @@
 This is particularly useful for the ``PanelFrontend`` but can be used by other frontends too.
 
 """
+
 from __future__ import annotations
 
 import os

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/frontend/panel/panel_frontend.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """The PanelFrontend wraps your Panel code in your LightningFlow."""
+
 from __future__ import annotations
 
 import inspect

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/frontend/panel/panel_serve_render_fn.py

@@ -28,6 +28,7 @@ Example:
         python panel_serve_render_fn
 
 """
+
 import inspect
 import os
 import pydoc

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/frontend/streamlit_base.py

@@ -16,6 +16,7 @@
 From here, we will call the render function that the user provided in ``configure_layout``.
 
 """
+
 import os
 import pydoc
 from typing import Callable

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/frontend/utils.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Utility functions for lightning Frontends."""
+
 from __future__ import annotations
 
 import inspect

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/launcher/launcher.py

@@ -66,12 +66,10 @@ def start_application_server(
     if isinstance(queues, Dict):
         kwargs.update(queues)
     else:
-        kwargs.update(
-            {
-                "api_publish_state_queue": queue_system.get_api_state_publish_queue(queue_id=queue_id),
-                "api_response_queue": queue_system.get_api_response_queue(queue_id=queue_id),
-            }
-        )
+        kwargs.update({
+            "api_publish_state_queue": queue_system.get_api_state_publish_queue(queue_id=queue_id),
+            "api_response_queue": queue_system.get_api_response_queue(queue_id=queue_id),
+        })
 
     app = load_app_from_file(entrypoint_file)
 

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/runners/multiprocess.py

@@ -125,14 +125,12 @@ class MultiProcessRuntime(Runtime):
                 # wait for server to be ready
                 has_started_queue.get()
 
-            if all(
-                [
-                    open_ui,
-                    "PYTEST_CURRENT_TEST" not in os.environ,
-                    not _is_headless(self.app),
-                    constants.LIGHTNING_CLOUDSPACE_HOST is None,
-                ]
-            ):
+            if all([
+                open_ui,
+                "PYTEST_CURRENT_TEST" not in os.environ,
+                not _is_headless(self.app),
+                constants.LIGHTNING_CLOUDSPACE_HOST is None,
+            ]):
                 click.launch(self._get_app_url())
 
             # Connect the runtime to the application.

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/testing/testing.py

@@ -338,9 +338,10 @@ def run_app_in_cloud(
         browser = p.chromium.launch(headless=bool(int(os.getenv("HEADLESS", "0"))))
         context = browser.new_context(
             # Eventually this will need to be deleted
-            http_credentials=HttpCredentials(
-                {"username": os.getenv("LAI_USER", "").strip(), "password": os.getenv("LAI_PASS", "")}
-            ),
+            http_credentials=HttpCredentials({
+                "username": os.getenv("LAI_USER", "").strip(),
+                "password": os.getenv("LAI_PASS", ""),
+            }),
             record_video_dir=os.path.join(_Config.video_location, TEST_APP_NAME),
             record_har_path=_Config.har_location,
         )

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/utilities/scheduler.py

@@ -46,13 +46,11 @@ class SchedulerThread(threading.Thread):
             if current_date > next_event:
                 component_delta = ComponentDelta(
                     id=metadata["name"],
-                    delta=Delta(
-                        {
-                            "values_changed": {
-                                f"root['calls']['scheduling']['{call_hash}']['running']": {"new_value": True}
-                            }
+                    delta=Delta({
+                        "values_changed": {
+                            f"root['calls']['scheduling']['{call_hash}']['running']": {"new_value": True}
                         }
-                    ),
+                    }),
                 )
                 self._app.delta_queue.put(component_delta)
                 metadata["start_time"] = next_event.isoformat()

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/utilities/tree.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Utilities for traversing the tree of components in an app."""
+
 from typing import TYPE_CHECKING, Type
 
 import lightning.app

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/app/utilities/types.py

@@ -24,5 +24,4 @@ ComponentTuple = (LightningFlow, LightningWork, Dict, List)
 
 @runtime_checkable
 class Hashable(Protocol):
-    def to_dict(self) -> t.Dict[str, t.Any]:
-        ...
+    def to_dict(self) -> t.Dict[str, t.Any]: ...

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/data/processing/dns.py

@@ -15,6 +15,7 @@ def optimize_dns_context(enable: bool) -> Any:
         optimize_dns(False)  # always disable the optimize DNS
         raise e
 
+
 def optimize_dns(enable: bool) -> None:
     if not _IS_IN_STUDIO:
         return
@@ -22,11 +23,14 @@ def optimize_dns(enable: bool) -> None:
     with open("/etc/resolv.conf") as f:
         lines = f.readlines()
 
-    if (
-        (enable and any("127.0.0.53" in line for line in lines))
-        or (not enable and any("127.0.0.1" in line for line in lines))
-    ): # noqa E501
-        Popen(f"sudo /home/zeus/miniconda3/envs/cloudspace/bin/python -c 'from lightning.data.processing.dns import _optimize_dns; _optimize_dns({enable})'", shell=True).wait() # noqa E501
+    if (enable and any("127.0.0.53" in line for line in lines)) or (
+        not enable and any("127.0.0.1" in line for line in lines)
+    ):  # E501
+        Popen(
+            f"sudo /home/zeus/miniconda3/envs/cloudspace/bin/python -c 'from lightning.data.processing.dns import _optimize_dns; _optimize_dns({enable})'",  # noqa E501
+            shell=True,
+        ).wait()
+
 
 def _optimize_dns(enable: bool) -> None:
     with open("/etc/resolv.conf") as f:
@@ -36,9 +40,9 @@ def _optimize_dns(enable: bool) -> None:
     for line in lines:
         if "nameserver 127" in line:
             if enable:
-                write_lines.append('nameserver 127.0.0.1\n')
+                write_lines.append("nameserver 127.0.0.1\n")
             else:
-                write_lines.append('nameserver 127.0.0.53\n')
+                write_lines.append("nameserver 127.0.0.53\n")
         else:
             write_lines.append(line)
 

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/data/processing/functions.py

@@ -187,7 +187,8 @@ def map(
     if not _IS_IN_STUDIO and (machine is not None or num_nodes is not None):
         raise ValueError(
             "Only https://lightning.ai/ supports multiple nodes or selecting a machine."
-            " Create an account to try it out.")
+            " Create an account to try it out."
+        )
 
     if not _IS_IN_STUDIO:
         print(

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/data/processing/image.py

@@ -8,6 +8,7 @@ _HTTPX_AVAILABLE = RequirementCache("httpx")
 # Credit to the https://github.com/rom1504/img2dataset Github repo
 # The code was taken from there. It has a MIT License.
 
+
 def _download_image(
     url: str,
     timeout: int = 10,

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/data/processing/readers.py

@@ -13,7 +13,6 @@ _PYARROW_AVAILABLE = RequirementCache("pyarrow")
 
 
 class BaseReader(ABC):
-
     def get_num_nodes(self) -> int:
         return int(os.getenv("DATA_OPTIMIZER_NUM_NODES", 1))
 
@@ -35,13 +34,13 @@ class BaseReader(ABC):
 @dataclass
 class ParquetSlice:
     """Keep track of a parquet file slice with its filepath, start and end."""
+
     filepath: str
     start: int
     end: int
 
 
 class ParquetReader(BaseReader):
-
     def __init__(self, num_rows: Optional[int] = 2048, to_pandas: bool = True) -> None:
         self.num_rows = num_rows
         self.to_pandas = to_pandas
@@ -52,12 +51,14 @@ class ParquetReader(BaseReader):
     def _get_num_rows(self, path: str) -> int:
         if _PYARROW_AVAILABLE:
             import pyarrow.dataset as ds
+
             df = ds.dataset(path).scanner()
             return df.count_rows()
 
         # FIXED: There is a bug in polars. This leads to read_parquet to hang.
         if _POLARS_AVAILABLE:
             import polars as pol
+
             df = pol.scan_parquet(path)
             num_rows = df.select(pol.len()).collect().item()
             return num_rows
@@ -67,6 +68,7 @@ class ParquetReader(BaseReader):
     def read(self, item: ParquetSlice) -> Any:
         if _POLARS_AVAILABLE:
             import polars as pol
+
             df = pol.scan_parquet(item.filepath).slice(item.start, item.end).collect()
 
             if self.to_pandas:
@@ -88,7 +90,6 @@ class ParquetReader(BaseReader):
 
         raise RuntimeError("Please, install either pyarrow or polars.")
 
-
     def items_to_workers(self, items: Any, num_workers: int) -> List[List[ParquetSlice]]:
         intervals = [(0, self._get_num_rows(item)) for item in items]
 
@@ -97,7 +98,8 @@ class ParquetReader(BaseReader):
 
         fake_distributed_env = _DistributedEnv(world_size, 0, self.get_num_nodes())
         parquet_indexes_per_worker, p_slices_per_worker = _associate_chunks_and_internals_to_ranks(
-            fake_distributed_env, list(range(len(items))), intervals, False)
+            fake_distributed_env, list(range(len(items))), intervals, False
+        )
 
         workers_user_items: List[List[ParquetSlice]] = [[] for _ in range(num_workers)]
 
@@ -111,9 +113,11 @@ class ParquetReader(BaseReader):
                 if self.num_rows:
                     workers_user_items[worker_idx % num_workers].extend([
                         ParquetSlice(
-                            items[parquet_index], p_slice_start, p_slice_start + self.num_rows
-                            if p_slice[1] > (p_slice_start + self.num_rows) else
-                            p_slice[1]
+                            items[parquet_index],
+                            p_slice_start,
+                            p_slice_start + self.num_rows
+                            if p_slice[1] > (p_slice_start + self.num_rows)
+                            else p_slice[1],
                         )
                         for parquet_index, p_slice in zip(parquet_indexes, p_slices)
                         for p_slice_start in range(p_slice[0], p_slice[1] + self.num_rows, self.num_rows)

{lightning-2.2.0.post0 → lightning-2.2.1}/src/lightning/data/streaming/cache.py

@@ -63,7 +63,8 @@ class Cache:
         if not _LIGHTNING_CLOUD_LATEST:
             raise ModuleNotFoundError(
                 "The `lightning-cloud` package in your environement is out-dated."
-                " Run: `pip install -U lightning-cloud` to resolve this.")
+                " Run: `pip install -U lightning-cloud` to resolve this."
+            )
 
         input_dir = _resolve_dir(input_dir)
         self._cache_dir = input_dir.path