Fix even more

Author: avandras

tests/test_citus.py

@@ -3,6 +3,7 @@
 import unittest
 
 from unittest.mock import Mock, patch, PropertyMock
+from typing import List
 from patroni.postgresql.mpp.citus import CitusHandler, PgDistNode, PgDistGroup
 from patroni.psycopg import ProgrammingError
 

tests/test_ctl.py

@@ -5,6 +5,7 @@
 import urllib3
 
 from datetime import datetime, timedelta
+from prettytable import PrettyTable
 from unittest import mock
 from unittest.mock import patch, Mock, PropertyMock
 from patroni import global_config