idea: context of optional dependency

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,