Ruff morning!

1 files changed, 6 insertions, 3 deletions

diff --git a/tests/managers/test_lucid.py b/tests/managers/test_lucid.py
index 654b58d..20dca22 100644
--- a/tests/managers/test_lucid.py
+++ b/tests/managers/test_lucid.py
@@ -1,6 +1,9 @@
+from __future__ import annotations
+
 import pytest
 
 from generalresearch.managers.lucid.profiling import get_profiling_library
+from generalresearch.pg_helper import PostgresConfig
 
 qids = ["42", "43", "45", "97", "120", "639", "15297"]
 
@@ -8,9 +11,9 @@ qids = ["42", "43", "45", "97", "120", "639", "15297"]
 class TestLucidProfiling:
 
     @pytest.mark.skip
-    def test_get_library(self, thl_web_rr):
+    def test_get_library(self, thl_web_rr: PostgresConfig):
         pks = [(qid, "us", "eng") for qid in qids]
-        qs = get_profiling_library(thl_web_rr: PostgresConfig, pks=pks)
+        qs = get_profiling_library(thl_web_rr, pks=pks)
         assert len(qids) == len(qs)
 
         # just making sure this doesn't raise errors
@@ -19,5 +22,5 @@ class TestLucidProfiling:
 
         # a lot will fail parsing because they have no options or the options are blank
         #   just asserting that we get some back
-        qs = get_profiling_library(thl_web_rr: PostgresConfig, country_iso="mx", language_iso="spa")
+        qs = get_profiling_library(thl_web_rr, country_iso="mx", language_iso="spa")
         assert len(qs) > 100