sorted imports

Author: ben-bay

data/build_training_data.py

@@ -1,5 +1,6 @@
 import csv
 import json
+
 from story.utils import *
 
 

data/make_reddit_data.py

@@ -1,7 +1,8 @@
 import json
-from story.utils import *
 import os
 
+from story.utils import *
+
 
 def load_stories(file):
 

data/scraper.py

@@ -1,7 +1,9 @@
+import json
+import time
+
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
-import time
-import json
+
 
 """
 format of tree is

generator/gpt2/download_model.py

@@ -1,5 +1,6 @@
 import os
 import sys
+
 import requests
 from tqdm import tqdm
 

generator/gpt2/gpt2_generator.py

@@ -1,14 +1,15 @@
-from story.utils import *
+import json
+import os
 import warnings
 
-warnings.filterwarnings("ignore")
-import os
+import numpy as np
 import tensorflow as tf
+from generator.gpt2.src import encoder, model, sample
+from story.utils import *
+
+warnings.filterwarnings("ignore")
 
 tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
-from generator.gpt2.src import sample, encoder, model
-import json
-import numpy as np
 
 
 class GPT2Generator:

generator/gpt2/src/encoder.py

@@ -1,10 +1,11 @@
 """Byte pair encoding utilities"""
 
-import os
 import json
-import regex as re
+import os
 from functools import lru_cache
 
+import regex as re
+
 
 @lru_cache()
 def bytes_to_unicode():

generator/gpt2/src/sample.py

@@ -1,5 +1,4 @@
 import tensorflow as tf
-
 from generator.gpt2.src import model
 
 

generator/simple/finetune.py

@@ -1,7 +1,7 @@
-import tarfile
 import os
-import gpt_2_simple as gpt2
+import tarfile
 
+import gpt_2_simple as gpt2
 
 model_name = "1558M"
 if not os.path.isdir(os.path.join("models", model_name)):

other/cacher.py

@@ -1,6 +1,7 @@
-from google.cloud import storage
 import os
 
+from google.cloud import storage
+
 
 class Cacher:
     def __init__(self, credentials_file, bucket_name="dungeon-cache"):

play.py

@@ -1,7 +1,10 @@
-from story.story_manager import *
+import os
+import sys
+import time
+
 from generator.gpt2.gpt2_generator import *
+from story.story_manager import *
 from story.utils import *
-import time, sys, os
 
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"