From 94af8e99563e1f38fbb4e411af6c529287b400ae Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Emel=20=C5=9Eim=C5=9Fek?= <emelhasdal@gmail.com>
Date: Thu, 6 Apr 2023 22:04:43 +0300
Subject: [PATCH] Fix upgrade tests

---
 src/test/regress/citus_tests/common.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/test/regress/citus_tests/common.py b/src/test/regress/citus_tests/common.py
index 609f13b61..0adbe3743 100644
--- 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