tensorflow
diff --git a/‎tfx_addons/copy_example_gen/component.py
Copy file name to clipboardExpand all lines: tfx_addons/copy_example_gen/component.py
+15-14Lines changed: 15 additions & 14 deletions b/‎tfx_addons/copy_example_gen/component.py
Copy file name to clipboardExpand all lines: tfx_addons/copy_example_gen/component.py
+15-14Lines changed: 15 additions & 14 deletions
@@ -46,25 +46,25 @@
 
 
 def _split_names_string_builder(split_names_list: List):
-
   """
   _split_names_string_builder() creates a string of split-names for input to
   output_example.split_names property.
 
   """
 
   str1 = "["
-  urlist_len = len(split_names_list)-1
+  urlist_len = len(split_names_list) - 1
   index = 0
 
   for element in split_names_list:
-    if(index==urlist_len):
-        str1 += "\""+element+"\""+"]"
+    if(index == urlist_len):
+        str1 += "\"" + element + "\"" + "]"
         break
-    str1 += "\""+element+"\""+","
-    index+=1
+    str1 += "\"" + element + "\"" + ","
+    index += 1
   return str1
 
+
 @component
 def CopyExampleGen(
     input_json_str: tfx.dsl.components.Parameter[str],
@@ -84,23 +84,24 @@ def CopyExampleGen(
   input_dict = json.loads(input_json_str)
 
   # Parse input_dict: creates a directory from the split-names and tfrecord uris provided
-  split_names=[]
+  split_names = []
   for key, value in input_dict.items():
     split_names.append(key)
 
-    split_names_string=_split_names_string_builder(split_names)
-    output_example.split_names=str(split_names_string)
+    split_names_string = _split_names_string_builder(split_names)
+    output_example.split_names = str(split_names_string)
 
     # Make directories
-    tfrecords_list=[]
-    output_example_uri=output_example.uri
+    tfrecords_list = []
+    output_example_uri = output_example.uri
 
     for key, value in input_dict.items():
       split_value=(f"/Split-{key}/")
       fileio.mkdir(f"{output_example_uri}{split_value}")
-      tfrecords_list=fileio.glob(f"{input_dict[key]}*.gz")
+      tfrecords_list = fileio.glob(f"{input_dict[key]}*.gz")
 
       # Copy files into directories
       for tfrecord in tfrecords_list:
-        file_name=os.path.basename(os.path.normpath(tfrecord))
-        fileio.copy(tfrecord, output_example.uri+split_value+file_name, True)
+        file_name = os.path.basename(os.path.normpath(tfrecord))
+        fileio.copy(tfrecord, output_example.uri + split_value + file_name,
+                    True)