Fix upgrade tests

2023-04-06 22:04:43 +03:00 · 2023-04-06 22:04:43 +03:00 · 94af8e9956
parent 9706df975d
commit 94af8e9956
1 changed files with 10 additions and 0 deletions
--- a/src/test/regress/citus_tests/common.py
+++ b/src/test/regress/citus_tests/common.py
@ -25,6 +25,16 @@ import utils
 from psycopg import sql
 from utils import USER

+from config import (  # noqa: E402
+    AFTER_CITUS_UPGRADE_COORD_SCHEDULE,
+    BEFORE_CITUS_UPGRADE_COORD_SCHEDULE,
+    CITUS_VERSION_SQL,
+    MASTER_VERSION,
+    MIXED_AFTER_CITUS_UPGRADE_SCHEDULE,
+    MIXED_BEFORE_CITUS_UPGRADE_SCHEDULE,
+    CitusUpgradeConfig,
+)
+
 LINUX = False
 MACOS = False
 FREEBSD = False