File size: 741 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
#
# Copyright (c) 2017-2021 NVIDIA CORPORATION. All rights reserved.
# This file is part of the WebDataset library.
# See the LICENSE file for licensing terms (BSD-style).
#
"""Mock implementations of torch interfaces when torch is not available."""
try:
from torch.utils.data import DataLoader, IterableDataset
except ModuleNotFoundError:
class IterableDataset:
"""Empty implementation of IterableDataset when torch is not available."""
class DataLoader:
"""Empty implementation of DataLoader when torch is not available."""
try:
from torch import Tensor as TorchTensor
except ModuleNotFoundError:
class TorchTensor:
"""Empty implementation of TorchTensor when torch is not available."""
|