Improve typing

aleph-im · Nov 14, 2024 · 0a4b2d6 · 0a4b2d6
1 parent 31c41ed
commit 0a4b2d6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/aleph/vm/orchestrator/custom_logs.py b/src/aleph/vm/orchestrator/custom_logs.py
@@ -3,9 +3,10 @@
 from contextvars import ContextVar
 
 from aleph.vm.models import VmExecution
+from aleph_message.models import ItemHash
 
-ctx_current_execution = ContextVar("current_execution")
-ctx_current_execution_hash = ContextVar("current_execution_hash")
+ctx_current_execution: ContextVar[VmExecution | None] = ContextVar("current_execution")
+ctx_current_execution_hash: ContextVar[ItemHash | None] = ContextVar("current_execution_hash")
 
 
 @contextlib.contextmanager