fix imports

Jong-Sig · Oct 31, 2019 · 960754d · 960754d
1 parent 1fa6b0c
commit 960754d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/firm_assign.py b/firm_assign.py
@@ -1,7 +1,7 @@
 import argparse
 import pandas as pd
-from standardize import standardize_strong
-from firm_tools import read_csv
+from tools.standardize import standardize_strong
+from tools.tables import read_csv
 
 # parse input arguments
 parser = argparse.ArgumentParser(description='USPTO assign fixer.')

diff --git a/firm_cites.py b/firm_cites.py
@@ -4,12 +4,12 @@
 import sqlite3
 import numpy as np
 import pandas as pd
-from firm_tools import read_csv
+from tools.tables import read_csv
 
 # parse input arguments
 parser = argparse.ArgumentParser(description='Merge patent citation data.')
 parser.add_argument('--output', type=str, default='tables', help='directory to operate on')
-parser.add_argument('--chunk', type=int, default=10_000_000, help='chunk size for citations')
+parser.add_argument('--chunk', type=int, default=10000000, help='chunk size for citations')
 args = parser.parse_args()
 
 # load in grant data

diff --git a/firm_merge.py b/firm_merge.py
@@ -1,7 +1,7 @@
 import numpy as np
 import pandas as pd
 from itertools import chain
-from firm_tools import read_csv
+from tools.tables import read_csv
 
 def merge_grants(output):
     print('Merging all grant data')