fix incorrect alias usage

CrayLabs · Aug 9, 2023 · 442d97e · 442d97e
1 parent 58c04e4
commit 442d97e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/smartsim/_core/utils/redis.py b/smartsim/_core/utils/redis.py
@@ -24,6 +24,8 @@
 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
+from __future__ import annotations
+
 import logging
 import redis
 import time
@@ -114,7 +116,7 @@ def check_cluster_status(
         # wait for cluster to spin up
         time.sleep(5)
         try:
-            redis_tester = _TRedisCluster(
+            redis_tester: _TRedisCluster = RedisCluster(
                 startup_nodes=cluster_nodes
             )
             redis_tester.set("__test__", "__test__")