From d9303e731ea38974d2228ea8cb0f0d0485c0b282 Mon Sep 17 00:00:00 2001
From: Zach Dwiel <zach.dwiel@intel.com>
Date: Tue, 20 Feb 2018 11:54:38 -0500
Subject: [PATCH] remove python2 compatibility

---
 architectures/tensorflow_components/architecture.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/architectures/tensorflow_components/architecture.py b/architectures/tensorflow_components/architecture.py
index 5cc77f7..41f7216 100644
--- a/architectures/tensorflow_components/architecture.py
+++ b/architectures/tensorflow_components/architecture.py
@@ -15,7 +15,6 @@
 #
 import time
 
-import six
 import numpy as np
 import tensorflow as tf
 
@@ -270,7 +269,7 @@ class TensorFlowArchitecture(Architecture):
     def _feed_dict(self, inputs):
         feed_dict = {}
         for input_name, input_value in inputs.items():
-            if isinstance(input_name, six.string_types):
+            if isinstance(input_name, str):
                 if input_name not in self.inputs:
                     raise ValueError((
                         'input name {input_name} was provided to create a feed '