Merge pull request #16 from diffgram/make-tensorflow-optional

Make tensorflow optional & fix setup imageio

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.tensorflow_diffgram.diffgram_tensorflow_dataset import DiffgramTensorflowDataset
 from diffgram.core.diffgram_dataset_iterator import DiffgramDatasetIterator
 from multiprocessing.pool import ThreadPool as Pool
 
@@ -166,6 +165,7 @@ def to_pytorch(self, transform = None):
         return pytorch_dataset
 
     def to_tensorflow(self):
+        from diffgram.tensorflow_diffgram.diffgram_tensorflow_dataset import DiffgramTensorflowDataset
         file_id_list = self.file_id_list
         diffgram_tensorflow_dataset = DiffgramTensorflowDataset(
             project = self.client,

sdk/requirements.txt

@@ -5,4 +5,4 @@ six>=1.9.0
 tensorflow>=1.12.0
 pillow
 torch
-imageio
+imageio>=2.9.0

sdk/setup.py

@@ -25,6 +25,7 @@
 	   'opencv-python>=4.0.0.21',
 	   'scipy>=1.1.0',
 	   'six>=1.9.0',
-	   'pillow>=6.1.0'
+	   'pillow>=6.1.0',
+       'imageio>=2.9.0'
 	]
 )