71

app/data/shakespeare_char/prepare.py

@@ -52,9 +52,8 @@ val_ids = np.array(val_ids, dtype=np.uint16)
 file_path_train = os.path.join(os.path.dirname(__file__), 'train.bin')
 file_path_val = os.path.join(os.path.dirname(__file__), 'val.bin')
 
-with open(file_path_train, 'wb') as file:
-
-with open(file_path_val, 'wb') as file:
+open(file_path_train, 'wb')
+open(file_path_val, 'wb')
 
 permissions = 0o755