diff --git a/mx_rec/saver/saver.py b/mx_rec/saver/saver.py index ffab7f7d26370782d5e023eea0d52bf4a94f838f..d776b699379e379047c3b14cd0de3281c21d1a9a 100644 --- a/mx_rec/saver/saver.py +++ b/mx_rec/saver/saver.py @@ -447,12 +447,8 @@ def write_binary_data(writing_path, suffix, data, attributes=None): target_data_dir = os.path.join(writing_path, data_file) with tf.io.gfile.GFile(target_data_dir, "ab") as file: - if check_file_system_is_hdfs(target_data_dir): - data = data.tostring() - file.write(data) - else: - data.astype(np.float32) - data.tofile(target_data_dir) + data = data.tostring() + file.write(data) def read_binary_data(reading_path: str, data_name: str, table_name: str, load_offset) -> dict: