Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
tchaton committed Apr 3, 2024
1 parent fe6e7b1 commit 3cf19bd
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions src/litdata/streaming/cache.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@

from litdata.constants import (
_INDEX_FILENAME,
_LIGHTNING_CLOUD_AVAILABLE,
_TORCH_GREATER_EQUAL_2_1_0,
)
from litdata.streaming.item_loader import BaseItemLoader
Expand Down Expand Up @@ -60,12 +59,6 @@ def __init__(
if not _TORCH_GREATER_EQUAL_2_1_0:
raise ModuleNotFoundError("PyTorch version 2.1 or higher is required to use the cache.")

if not _LIGHTNING_CLOUD_AVAILABLE:
raise ModuleNotFoundError(
"The `lightning-cloud` package in your environement is out-dated."
" Run: `pip install -U lightning-cloud` to resolve this."
)

input_dir = _resolve_dir(input_dir)
self._cache_dir = input_dir.path
assert self._cache_dir
Expand Down

0 comments on commit 3cf19bd

Please sign in to comment.