Skip to content

Commit 0a4b2d6

Browse files
committed
Improve typing
1 parent 31c41ed commit 0a4b2d6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/aleph/vm/orchestrator/custom_logs.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,10 @@
33
from contextvars import ContextVar
44

55
from aleph.vm.models import VmExecution
6+
from aleph_message.models import ItemHash
67

7-
ctx_current_execution = ContextVar("current_execution")
8-
ctx_current_execution_hash = ContextVar("current_execution_hash")
8+
ctx_current_execution: ContextVar[VmExecution | None] = ContextVar("current_execution")
9+
ctx_current_execution_hash: ContextVar[ItemHash | None] = ContextVar("current_execution_hash")
910

1011

1112
@contextlib.contextmanager

0 commit comments

Comments
 (0)