Merge pull request #12 from diffgram/make-py-torch-optional

make pytorch optional

Author: anthony-chaudhary

sdk/diffgram/core/directory.py

@@ -1,7 +1,6 @@
 from diffgram.file.file import File
 from ..regular.regular import refresh_from_dict
 import logging
-from diffgram.pytorch_diffgram.diffgram_pytorch_dataset import DiffgramPytorchDataset
 from diffgram.tensorflow_diffgram.diffgram_tensorflow_dataset import DiffgramTensorflowDataset
 from diffgram.core.diffgram_dataset_iterator import DiffgramDatasetIterator
 from multiprocessing.pool import ThreadPool as Pool
@@ -155,6 +154,7 @@ def to_pytorch(self, transform = None):
             Transforms the file list inside the dataset into a pytorch dataset.
         :return:
         """
+        from diffgram.pytorch_diffgram.diffgram_pytorch_dataset import DiffgramPytorchDataset
         file_id_list = self.file_id_list
         pytorch_dataset = DiffgramPytorchDataset(
             project = self.client,

sdk/diffgram/core/sliced_directory.py

@@ -1,5 +1,4 @@
 from diffgram.core.directory import Directory
-from diffgram.pytorch_diffgram.diffgram_pytorch_dataset import DiffgramPytorchDataset
 from diffgram.tensorflow_diffgram.diffgram_tensorflow_dataset import DiffgramTensorflowDataset
 import urllib
 
@@ -37,7 +36,7 @@ def to_pytorch(self, transform = None):
             Transforms the file list inside the dataset into a pytorch dataset.
         :return:
         """
-
+        from diffgram.pytorch_diffgram.diffgram_pytorch_dataset import DiffgramPytorchDataset
         pytorch_dataset = DiffgramPytorchDataset(
             project = self.client,
             diffgram_file_id_list = self.file_id_list,

sdk/diffgram/pytorch_diffgram/diffgram_pytorch_dataset.py

@@ -1,7 +1,12 @@
 from torch.utils.data import Dataset, DataLoader
-import torch as torch  # type: ignore
 from diffgram.core.diffgram_dataset_iterator import DiffgramDatasetIterator
 
+try:
+    import torch as torch  # type: ignore
+except ModuleNotFoundError:
+    raise ModuleNotFoundError(
+        "'torch' module should be installed to convert the Dataset into torch (pytorch) format"
+    )
 
 class DiffgramPytorchDataset(DiffgramDatasetIterator, Dataset):