Import order fix and update to patch detection.

Dallinger · Jan 13, 2025 · f17bb5f · f17bb5f
1 parent a2bf479
commit f17bb5f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/dlgr/griduniverse/gevent_patch.py b/dlgr/griduniverse/gevent_patch.py
@@ -6,11 +6,12 @@
 # isort: on
 
 # This definitely belongs in dallinger.data
-import psycopg2  # noqa
 from contextlib import contextmanager  # noqa
+
+import psycopg2  # noqa
 from dallinger import data  # noqa
 
-is_patched = getattr(data, "_patched", False)
+is_patched = getattr(data, "_paused_thread", False)
 
 
 @contextmanager
@@ -30,5 +31,5 @@ def _patched_copy_db_to_csv(dsn, path, scrub_pii=False):
         with _paused_thread():
             return _orig_copy_db_to_csv(dsn, path, scrub_pii)
 
-    data._patched = True
+    data._paused_thread = _paused_thread
     data.copy_db_to_csv = _patched_copy_db_to_csv