@@ -376,7 +376,7 @@ def connect(self, *args, **kwargs):
376
376
# handles the case that source is specified
377
377
# with a function
378
378
sourcename = source [0 ]
379
- elif isinstance (source , six . string_types ):
379
+ elif isinstance (source , string_types ):
380
380
sourcename = source
381
381
else :
382
382
raise Exception (('Unknown source specification in '
@@ -397,7 +397,7 @@ def connect(self, *args, **kwargs):
397
397
# turn functions into strings
398
398
for srcnode , destnode , connects in connection_list :
399
399
for idx , (src , dest ) in enumerate (connects ):
400
- if isinstance (src , tuple ) and not isinstance (src [1 ], six . string_types ):
400
+ if isinstance (src , tuple ) and not isinstance (src [1 ], string_types ):
401
401
function_source = getsource (src [1 ])
402
402
connects [idx ] = ((src [0 ], function_source , src [2 :]), dest )
403
403
@@ -918,7 +918,7 @@ def _set_input(self, object, name, newvalue):
918
918
919
919
def _set_node_input (self , node , param , source , sourceinfo ):
920
920
"""Set inputs of a node given the edge connection"""
921
- if isinstance (sourceinfo , six . string_types ):
921
+ if isinstance (sourceinfo , string_types ):
922
922
val = source .get_output (sourceinfo )
923
923
elif isinstance (sourceinfo , tuple ):
924
924
if callable (sourceinfo [1 ]):
0 commit comments