File tree 1 file changed +7
-7
lines changed
Filter options
tfx_addons/copy_example_gen
1 file changed +7
-7
lines changed
Original file line number Diff line number Diff line change @@ -57,9 +57,9 @@ def _split_names_string_builder(split_names_list: List):
57
57
index = 0
58
58
59
59
for element in split_names_list :
60
- if (index == urlist_len ):
61
- str1 += "\" " + element + "\" " + "]"
62
- break
60
+ if (index == urlist_len ):
61
+ str1 += "\" " + element + "\" " + "]"
62
+ break
63
63
str1 += "\" " + element + "\" " + ","
64
64
index += 1
65
65
return str1
@@ -79,8 +79,8 @@ def CopyExampleGen(
79
79
Following the creation of the `Split-name` folder, the files in the uri path will then be copied
80
80
into the designated `Split-name` folder.
81
81
82
- """
83
-
82
+ """
83
+
84
84
input_dict = json .loads (input_json_str )
85
85
86
86
# Parse input_dict: creates a directory from the split-names and tfrecord uris provided
@@ -90,13 +90,13 @@ def CopyExampleGen(
90
90
91
91
split_names_string = _split_names_string_builder (split_names )
92
92
output_example .split_names = str (split_names_string )
93
-
93
+
94
94
# Make directories
95
95
tfrecords_list = []
96
96
output_example_uri = output_example .uri
97
97
98
98
for key , value in input_dict .items ():
99
- split_value = (f"/Split-{ key } /" )
99
+ split_value = (f"/Split-{ key } /" )
100
100
fileio .mkdir (f"{ output_example_uri } { split_value } " )
101
101
tfrecords_list = fileio .glob (f"{ input_dict [key ]} *.gz" )
102
102
You can’t perform that action at this time.
0 commit comments