Merge branch 'master' into dev-greg

1 files changed, 8 insertions, 1 deletions

diff --git a/tests/managers/test_lucid.py b/tests/managers/test_lucid.py
index 1a1bae7..6771a0c 100644
--- a/tests/managers/test_lucid.py
+++ b/tests/managers/test_lucid.py
@@ -1,14 +1,21 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 import pytest
 
 from generalresearch.managers.lucid.profiling import get_profiling_library
 
+if TYPE_CHECKING:
+    from generalresearch.pg_helper import PostgresConfig
+
 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, pks=pks)
         assert len(qids) == len(qs)