Expose error more nicely

4 files changed, 2 insertions, 5 deletions

diff --git a/synapse/storage/__init__.py b/synapse/storage/__init__.py

index 4be629bff8..48a0633746 100644
--- a/synapse/storage/__init__.py
+++ b/synapse/storage/__init__.py
@@ -41,9 +41,6 @@ from .end_to_end_keys import EndToEndKeyStore
 
 from .receipts import ReceiptsStore
 
-from ._schema_prepare import UpgradeDatabaseException
-
-__all__ = [UpgradeDatabaseException]
 
 import logging
 
diff --git a/synapse/storage/engines/postgres.py b/synapse/storage/engines/postgres.py

index 949396044e..7e45dabf4c 100644
--- a/synapse/storage/engines/postgres.py
+++ b/synapse/storage/engines/postgres.py
@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from synapse.storage._schema_prepare import prepare_database
+from synapse.storage.schema_prepare import prepare_database
 
 from ._base import IncorrectDatabaseSetup
 
diff --git a/synapse/storage/engines/sqlite3.py b/synapse/storage/engines/sqlite3.py

index a66815ef2d..0eeaa45d19 100644
--- a/synapse/storage/engines/sqlite3.py
+++ b/synapse/storage/engines/sqlite3.py
@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from synapse.storage._schema_prepare import (
+from synapse.storage.schema_prepare import (
     prepare_database, prepare_sqlite3_database
 )
 
diff --git a/synapse/storage/_schema_prepare.py b/synapse/storage/schema_prepare.py

index 1ddf55be4d..1ddf55be4d 100644
--- a/synapse/storage/_schema_prepare.py
+++ b/synapse/storage/schema_prepare.py