diff --git a/pgdb/__init__.py b/pgdb/__init__.py index 7879059..6613eac 100644 --- a/pgdb/__init__.py +++ b/pgdb/__init__.py @@ -13,4 +13,4 @@ def connect(url=None, schema=None, sql_path='sql', multiprocessing=False): """ if url is None: url = os.environ.get('DATABASE_URL') - return Database(url, schema) + return Database(url, schema, multiprocessing=multiprocessing) diff --git a/setup.py b/setup.py index 544f8ca..400aec7 100644 --- a/setup.py +++ b/setup.py @@ -50,6 +50,6 @@ }, entry_points=""" [console_scripts] - pgdb=pgdb.scripts.cli:cli + pgdb=pgdb.cli:cli """ )