lightning 2.2.0__tar.gz → 2.2.0.post0__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning
-Version: 2.2.0
+Version: 2.2.0.post0
 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.
@@ -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">Docs</a> •
+  <a href="https://pytorch-lightning.readthedocs.io/en/2.2.0.post0">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)](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.0.post0)](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 → lightning-2.2.0.post0}/requirements/pytorch/examples.txt

@@ -1,6 +1,7 @@
 # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
 #  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
 
+requests <2.32.0
 torchvision >=0.14.0, <0.17.0
 gym[classic_control] >=0.17.0, <0.27.0
 ipython[all] <8.15.0

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/__setup__.py

@@ -114,7 +114,9 @@ def _setup_args() -> Dict[str, Any]:
         "python_requires": ">=3.8",  # todo: take the lowes based on all packages
         "entry_points": {
             "console_scripts": [
-                "lightning = lightning:_cli_entry_point",
+                "fabric = lightning.fabric.cli:_main",
+                "lightning = lightning.fabric.cli:_legacy_main",
+                "lightning_app = lightning:_cli_entry_point",
             ],
         },
         "setup_requires": [],

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/app/CHANGELOG.md

@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
+## [2.2.0] - 2024-02-09
+
+## Changed
+
+- Renames the `lightning` cli to `lightning_app` ([#19440](https://github.com/Lightning-AI/pytorch-lightning/pull/19440))
+
 
 ## [2.1.4] - 2024-01-31
 

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/README.md

@@ -3,7 +3,7 @@
 This ⚡ [Lightning app](https://lightning.ai/) ⚡ was generated automatically with:
 
 ```bash
-lightning init app placeholdername
+lightning_app init app placeholdername
 ```
 
 ## To run placeholdername
@@ -11,19 +11,19 @@ lightning init app placeholdername
 First, install placeholdername (warning: this app has not been officially approved on the lightning gallery):
 
 ```bash
-lightning install app https://github.com/theUser/placeholdername
+lightning_app install app https://github.com/theUser/placeholdername
 ```
 
 Once the app is installed, run it locally with:
 
 ```bash
-lightning run app placeholdername/app.py
+lightning_app run app placeholdername/app.py
 ```
 
 Run it on the [lightning cloud](https://lightning.ai/) with:
 
 ```bash
-lightning run app placeholdername/app.py --cloud
+lightning_app run app placeholdername/app.py --cloud
 ```
 
 ## to test and link

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/app/cli/cmd_react_ui_init.py

@@ -59,7 +59,7 @@ def _copy_and_setup_react_ui(dest_dir: Optional[str] = None) -> None:
             return la.frontend.StaticWebFrontend(Path(__file__).parent / "react-ui/src/dist")
 
     ⚡ run the example_app.py to see it live!
-    lightning run app {dest_dir}/example_app.py
+    lightning_app run app {dest_dir}/example_app.py
 
     """
     logger.info(m)
@@ -80,7 +80,7 @@ def _check_react_prerequisites() -> None:
 
     if not has_npm:
         m = """
-        This machine is missing 'npm'. Please install npm and rerun 'lightning init react-ui' again.
+        This machine is missing 'npm'. Please install npm and rerun 'lightning_app init react-ui' again.
 
         Install instructions: https://docs.npmjs.com/downloading-and-installing-node-js-and-npm
         """
@@ -94,7 +94,7 @@ def _check_react_prerequisites() -> None:
 
     if not has_node:
         m = """
-        This machine is missing 'node'. Please install node and rerun 'lightning init react-ui' again.
+        This machine is missing 'node'. Please install node and rerun 'lightning_app init react-ui' again.
 
         Install instructions: https://docs.npmjs.com/downloading-and-installing-node-js-and-npm
         """

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/app/cli/commands/app_commands.py

@@ -61,7 +61,7 @@ def _run_app_command(app_name: str, app_id: Optional[str]):
                 if value == command and "_" in value:
                     print(
                         f"The command `{value}` was provided with an underscore and it isn't allowed."
-                        f"Instead, use `lightning {value.replace('_', ' ')}`."
+                        f"Instead, use `lightning_app {value.replace('_', ' ')}`."
                     )
                     sys.exit(0)
             break
@@ -88,7 +88,7 @@ def _run_app_command(app_name: str, app_id: Optional[str]):
 def _handle_command_without_client(command: str, metadata: Dict, url: str) -> None:
     supported_params = list(metadata["parameters"])
     if _is_running_help(sys.argv):
-        print(f"Usage: lightning {command} [ARGS]...")
+        print(f"Usage: lightning_app {command} [ARGS]...")
         print(" ")
         print("Options")
         for param in supported_params:

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/app/cli/commands/ls.py

@@ -81,7 +81,7 @@ def ls(path: Optional[str] = None, print: bool = True, use_live: bool = True) ->
         # This happens if the user changes cluster and the project doesn't exit.
         if len(project) == 0:
             return _error_and_exit(
-                f"There isn't any Lightning Project matching the name {splits[0]}." " HINT: Use `lightning cd`."
+                f"There isn't any Lightning Project matching the name {splits[0]}." " HINT: Use `lightning_app cd`."
             )
 
         project_id = project[0].project_id

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/README.md

@@ -3,7 +3,7 @@
 This ⚡ [Lightning component](https://lightning.ai/) ⚡ was generated automatically with:
 
 ```bash
-lightning init component placeholdername
+lightning_app init component placeholdername
 ```
 
 ## To run placeholdername
@@ -11,7 +11,7 @@ lightning init component placeholdername
 First, install placeholdername (warning: this component has not been officially approved on the lightning gallery):
 
 ```bash
-lightning install component https://github.com/theUser/placeholdername
+lightning_app install component https://github.com/theUser/placeholdername
 ```
 
 Once the app is installed, use it in an app:

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/setup.py

@@ -5,7 +5,7 @@ from setuptools import find_packages, setup
 setup(
     name="placeholdername",
     version="0.0.0",
-    description="⚡ Lightning component ⚡ generated with command: lightning init component",
+    description="⚡ Lightning component ⚡ generated with command: lightning_app init component",
     author="",
     author_email="",
     # REPLACE WITH YOUR OWN GITHUB PROJECT LINK

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/app/cli/connect/app.py

@@ -129,11 +129,11 @@ def connect_app(app_name_or_id: str):
         with open(connected_file, "w") as f:
             f.write(app_name_or_id + "\n")
 
-        click.echo("The lightning CLI now responds to app commands. Use 'lightning --help' to see them.")
+        click.echo("The lightning App CLI now responds to app commands. Use 'lightning_app --help' to see them.")
         click.echo(" ")
 
         Popen(
-            f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning --help",
+            f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning_app --help",
             shell=True,
             stdout=sys.stdout,
             stderr=sys.stderr,
@@ -147,11 +147,11 @@ def connect_app(app_name_or_id: str):
             shutil.copytree(matched_commands, commands)
             shutil.copy(matched_connected_file, connected_file)
 
-        click.echo("The lightning CLI now responds to app commands. Use 'lightning --help' to see them.")
+        click.echo("The lightning App CLI now responds to app commands. Use 'lightning_app --help' to see them.")
         click.echo(" ")
 
         Popen(
-            f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning --help",
+            f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning_app --help",
             shell=True,
             stdout=sys.stdout,
             stderr=sys.stderr,
@@ -205,11 +205,11 @@ def connect_app(app_name_or_id: str):
             f.write(retriever.app_name + "\n")
             f.write(retriever.app_id + "\n")
 
-        click.echo("The lightning CLI now responds to app commands. Use 'lightning --help' to see them.")
+        click.echo("The lightning App CLI now responds to app commands. Use 'lightning_app --help' to see them.")
         click.echo(" ")
 
         Popen(
-            f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning --help",
+            f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning_app --help",
             shell=True,
             stdout=sys.stdout,
             stderr=sys.stderr,
@@ -238,7 +238,7 @@ def disconnect_app(logout: bool = False):
         if not logout:
             click.echo(
                 "You aren't connected to any Lightning App. "
-                "Please use `lightning connect app_name_or_id` to connect to one."
+                "Please use `lightning_app connect app_name_or_id` to connect to one."
             )
 
 

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/app/cli/lightning_cli.py

@@ -104,7 +104,7 @@ def main() -> None:
                     _run_app_command(app_name, app_id)
 
                 click.echo()
-                click.echo(message + " Return to the primary CLI with `lightning disconnect`.")
+                click.echo(message + " Return to the primary CLI with `lightning_app disconnect`.")
         else:
             _main()
 

{lightning-2.2.0 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/README.md

@@ -5,7 +5,7 @@ This is a full react template ready to use in a component
 This UI was automatically generated with:
 
 ```commandline
-lightning init react-ui
+lightning_app init react-ui
 ```
 
 ### Delete files
@@ -26,7 +26,7 @@ This template comes with `example_app.py` to show how to integrate the UI into a
 run it with:
 
 ```bash
-lightning run app react-ui/example_app.py
+lightning_app run app react-ui/example_app.py
 ```
 
 ### Connect React to your component

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

@@ -716,11 +716,11 @@ class LightningFlow:
 
         .. code-block:: bash
 
-            lightning run app app.py
+            lightning_app run app app.py
 
         .. code-block:: bash
 
-            lightning my_command_name --args name=my_own_name
+            lightning_app my_command_name --args name=my_own_name
 
         """
         raise NotImplementedError

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

@@ -131,7 +131,7 @@ class CloudRuntime(Runtime):
             user = self.backend.client.auth_service_get_user()
             if not user.features.code_tab:
                 rich.print(
-                    "[red]The `lightning open` command has not been enabled for your account. "
+                    "[red]The `lightning_app open` command has not been enabled for your account. "
                     "To request access, please contact support@lightning.ai[/red]"
                 )
                 sys.exit(1)

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

@@ -275,7 +275,7 @@ def run_app_in_cloud(
     token = res.json()["token"]
 
     # 3. Disconnect from the App if any.
-    Popen("lightning logout", shell=True).wait()
+    Popen("lightning_app logout", shell=True).wait()
 
     # 4. Launch the application in the cloud from the Lightning CLI.
     with tempfile.TemporaryDirectory() as tmpdir:

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

@@ -239,7 +239,7 @@ def _patch_sys_argv():
     """This function modifies the ``sys.argv`` by extracting the arguments after ``--app_args`` and removed everything
     else before executing the user app script.
 
-    The command: ``lightning run app app.py --without-server --app_args --use_gpu --env ...`` will be converted into
+    The command: ``lightning_app run app app.py --without-server --app_args --use_gpu --env ...`` will be converted into
     ``app.py --use_gpu``
 
     """

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

@@ -144,7 +144,7 @@ class Auth:
 
         raise ValueError(
             "We couldn't find any credentials linked to your account. "
-            "Please try logging in using the CLI command `lightning login`"
+            "Please try logging in using the CLI command `lightning_app login`"
         )
 
 

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

@@ -1,7 +1,7 @@
+from lightning.data.processing.functions import map, optimize, walk
 from lightning.data.streaming.combined import CombinedStreamingDataset
 from lightning.data.streaming.dataloader import StreamingDataLoader
 from lightning.data.streaming.dataset import StreamingDataset
-from lightning.data.streaming.functions import map, optimize, walk
 
 __all__ = [
     "LightningDataset",

{lightning-2.2.0/src/lightning/data/streaming → lightning-2.2.0.post0/src/lightning/data/processing}/data_processor.py

@@ -20,11 +20,7 @@ import numpy as np
 from tqdm.auto import tqdm as _tqdm
 
 from lightning import seed_everything
-from lightning.data.processing.readers import BaseReader
-from lightning.data.streaming import Cache
-from lightning.data.streaming.cache import Dir
-from lightning.data.streaming.client import S3Client
-from lightning.data.streaming.constants import (
+from lightning.data.constants import (
     _BOTO3_AVAILABLE,
     _DEFAULT_FAST_DEV_RUN_ITEMS,
     _INDEX_FILENAME,
@@ -32,6 +28,10 @@ from lightning.data.streaming.constants import (
     _LIGHTNING_CLOUD_LATEST,
     _TORCH_GREATER_EQUAL_2_1_0,
 )
+from lightning.data.processing.readers import BaseReader
+from lightning.data.streaming import Cache
+from lightning.data.streaming.cache import Dir
+from lightning.data.streaming.client import S3Client
 from lightning.data.streaming.resolver import _resolve_dir
 from lightning.data.utilities.broadcast import broadcast_object
 from lightning.data.utilities.packing import _pack_greedily

lightning-2.2.0.post0/src/lightning/data/processing/dns.py

@@ -0,0 +1,47 @@
+from contextlib import contextmanager
+from subprocess import Popen
+from typing import Any
+
+from lightning.data.constants import _IS_IN_STUDIO
+
+
+@contextmanager
+def optimize_dns_context(enable: bool) -> Any:
+    optimize_dns(enable)
+    try:
+        yield
+        optimize_dns(False)  # always disable the optimize DNS
+    except Exception as e:
+        optimize_dns(False)  # always disable the optimize DNS
+        raise e
+
+def optimize_dns(enable: bool) -> None:
+    if not _IS_IN_STUDIO:
+        return
+
+    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
+
+def _optimize_dns(enable: bool) -> None:
+    with open("/etc/resolv.conf") as f:
+        lines = f.readlines()
+
+    write_lines = []
+    for line in lines:
+        if "nameserver 127" in line:
+            if enable:
+                write_lines.append('nameserver 127.0.0.1\n')
+            else:
+                write_lines.append('nameserver 127.0.0.53\n')
+        else:
+            write_lines.append(line)
+
+    with open("/etc/resolv.conf", "w") as f:
+        for line in write_lines:
+            f.write(line)

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

@@ -22,9 +22,10 @@ from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union
 
 import torch
 
+from lightning.data.constants import _IS_IN_STUDIO, _TORCH_GREATER_EQUAL_2_1_0
+from lightning.data.processing.data_processor import DataChunkRecipe, DataProcessor, DataTransformRecipe
+from lightning.data.processing.dns import optimize_dns_context
 from lightning.data.processing.readers import BaseReader
-from lightning.data.streaming.constants import _IS_IN_STUDIO, _TORCH_GREATER_EQUAL_2_1_0
-from lightning.data.streaming.data_processor import DataChunkRecipe, DataProcessor, DataTransformRecipe
 from lightning.data.streaming.resolver import (
     Dir,
     _assert_dir_has_index_file,
@@ -218,7 +219,8 @@ def map(
             weights=weights,
             reader=reader,
         )
-        return data_processor.run(LambdaDataTransformRecipe(fn, inputs))
+        with optimize_dns_context(True):
+            return data_processor.run(LambdaDataTransformRecipe(fn, inputs))
     return _execute(
         f"data-prep-map-{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}",
         num_nodes,
@@ -303,15 +305,18 @@ def optimize(
             reorder_files=reorder_files,
             reader=reader,
         )
-        return data_processor.run(
-            LambdaDataChunkRecipe(
-                fn,
-                inputs,
-                chunk_size=chunk_size,
-                chunk_bytes=chunk_bytes,
-                compression=compression,
+
+        with optimize_dns_context(True):
+            data_processor.run(
+                LambdaDataChunkRecipe(
+                    fn,
+                    inputs,
+                    chunk_size=chunk_size,
+                    chunk_bytes=chunk_bytes,
+                    compression=compression,
+                )
             )
-        )
+        return None
     return _execute(
         f"data-prep-optimize-{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}",
         num_nodes,

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

@@ -5,8 +5,8 @@ from typing import Any, List, Optional
 
 from lightning_utilities.core.imports import RequirementCache
 
-from lightning.data.streaming.shuffle import _associate_chunks_and_internals_to_ranks
 from lightning.data.utilities.env import _DistributedEnv
+from lightning.data.utilities.shuffle import _associate_chunks_and_internals_to_ranks
 
 _POLARS_AVAILABLE = RequirementCache("polars")
 _PYARROW_AVAILABLE = RequirementCache("pyarrow")

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

@@ -13,18 +13,14 @@
 
 from lightning.data.streaming.cache import Cache
 from lightning.data.streaming.combined import CombinedStreamingDataset
-from lightning.data.streaming.data_processor import DataChunkRecipe, DataProcessor, DataTransformRecipe
 from lightning.data.streaming.dataloader import StreamingDataLoader
 from lightning.data.streaming.dataset import StreamingDataset
 from lightning.data.streaming.item_loader import TokensLoader
 
 __all__ = [
     "Cache",
-    "DataProcessor",
     "StreamingDataset",
     "CombinedStreamingDataset",
     "StreamingDataLoader",
-    "DataTransformRecipe",
-    "DataChunkRecipe",
     "TokensLoader",
 ]

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

@@ -15,7 +15,7 @@ import logging
 import os
 from typing import Any, Dict, List, Optional, Tuple, Union
 
-from lightning.data.streaming.constants import (
+from lightning.data.constants import (
     _INDEX_FILENAME,
     _LIGHTNING_CLOUD_LATEST,
     _TORCH_GREATER_EQUAL_2_1_0,

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

@@ -2,7 +2,7 @@ import os
 from time import time
 from typing import Any, Optional
 
-from lightning.data.streaming.constants import _BOTO3_AVAILABLE
+from lightning.data.constants import _BOTO3_AVAILABLE
 
 if _BOTO3_AVAILABLE:
     import boto3

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

@@ -15,7 +15,7 @@ import json
 import os
 from typing import Any, Dict, List, Optional, Tuple
 
-from lightning.data.streaming.constants import _INDEX_FILENAME, _TORCH_GREATER_EQUAL_2_1_0
+from lightning.data.constants import _INDEX_FILENAME, _TORCH_GREATER_EQUAL_2_1_0
 from lightning.data.streaming.downloader import get_downloader_cls
 from lightning.data.streaming.item_loader import BaseItemLoader, PyTreeLoader, TokensLoader
 from lightning.data.streaming.sampler import ChunkedIndex

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

@@ -33,13 +33,13 @@ from torch.utils.data.dataloader import (
 )
 from torch.utils.data.sampler import BatchSampler, Sampler
 
+from lightning.data.constants import _DEFAULT_CHUNK_BYTES, _TORCH_GREATER_EQUAL_2_1_0, _VIZ_TRACKER_AVAILABLE
 from lightning.data.streaming import Cache
 from lightning.data.streaming.combined import (
     __NUM_SAMPLES_YIELDED_KEY__,
     __SAMPLES_KEY__,
     CombinedStreamingDataset,
 )
-from lightning.data.streaming.constants import _DEFAULT_CHUNK_BYTES, _TORCH_GREATER_EQUAL_2_1_0, _VIZ_TRACKER_AVAILABLE
 from lightning.data.streaming.dataset import StreamingDataset
 from lightning.data.streaming.sampler import CacheBatchSampler
 from lightning.data.utilities.env import _DistributedEnv

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

@@ -19,11 +19,11 @@ from typing import Any, Dict, List, Optional, Tuple, Union
 import numpy as np
 from torch.utils.data import IterableDataset
 
-from lightning.data.streaming import Cache
-from lightning.data.streaming.constants import (
+from lightning.data.constants import (
     _DEFAULT_CACHE_DIR,
     _INDEX_FILENAME,
 )
+from lightning.data.streaming import Cache
 from lightning.data.streaming.item_loader import BaseItemLoader
 from lightning.data.streaming.resolver import Dir, _resolve_dir
 from lightning.data.streaming.sampler import ChunkedIndex

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

@@ -19,8 +19,8 @@ from urllib import parse
 
 from filelock import FileLock, Timeout
 
+from lightning.data.constants import _INDEX_FILENAME
 from lightning.data.streaming.client import S3Client
-from lightning.data.streaming.constants import _INDEX_FILENAME
 
 
 class Downloader(ABC):

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

@@ -19,7 +19,7 @@ from typing import Any, Dict, List, Optional, Tuple
 import numpy as np
 import torch
 
-from lightning.data.streaming.constants import (
+from lightning.data.constants import (
     _TORCH_DTYPES_MAPPING,
     _TORCH_GREATER_EQUAL_2_1_0,
 )

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

@@ -20,8 +20,8 @@ from queue import Empty
 from threading import Thread
 from typing import Any, Dict, List, Optional, Tuple, Union
 
+from lightning.data.constants import _TORCH_GREATER_EQUAL_2_1_0
 from lightning.data.streaming.config import ChunksConfig
-from lightning.data.streaming.constants import _TORCH_GREATER_EQUAL_2_1_0
 from lightning.data.streaming.item_loader import BaseItemLoader, PyTreeLoader
 from lightning.data.streaming.sampler import ChunkedIndex
 from lightning.data.streaming.serializers import Serializer, _get_serializers

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

@@ -23,7 +23,7 @@ import numpy as np
 import torch
 from lightning_utilities.core.imports import RequirementCache
 
-from lightning.data.streaming.constants import _NUMPY_DTYPES_MAPPING, _TORCH_DTYPES_MAPPING
+from lightning.data.constants import _NUMPY_DTYPES_MAPPING, _TORCH_DTYPES_MAPPING
 
 _PIL_AVAILABLE = RequirementCache("PIL")
 _TORCH_VISION_AVAILABLE = RequirementCache("torchvision")