From f777b3406e25fef5a91073f22b29eddb87e5476e Mon Sep 17 00:00:00 2001
From: Anthony Sarkis <18080164+anthony-sarkis@users.noreply.github.com>
Date: Thu, 23 Sep 2021 09:38:11 -0700
Subject: [PATCH 1/2] make optional, similar to pytorch

---
 sdk/diffgram/core/directory.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdk/diffgram/core/directory.py b/sdk/diffgram/core/directory.py
index 54d1a2a..ee42ba3 100644
--- a/sdk/diffgram/core/directory.py
+++ b/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,

From b2a069214f440d11f063f571d6a00166d5a8cfbc Mon Sep 17 00:00:00 2001
From: Anthony Sarkis <18080164+anthony-sarkis@users.noreply.github.com>
Date: Thu, 23 Sep 2021 10:34:42 -0700
Subject: [PATCH 2/2] add imageio requirement to setup

---
 sdk/requirements.txt | 2 +-
 sdk/setup.py         | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/sdk/requirements.txt b/sdk/requirements.txt
index 02704a9..651c49d 100644
--- a/sdk/requirements.txt
+++ b/sdk/requirements.txt
@@ -5,4 +5,4 @@ six>=1.9.0
 tensorflow>=1.12.0
 pillow
 torch
-imageio
\ No newline at end of file
+imageio>=2.9.0
\ No newline at end of file
diff --git a/sdk/setup.py b/sdk/setup.py
index 9702a09..e7df02a 100644
--- a/sdk/setup.py
+++ b/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'
 	]
 )
\ No newline at end of file