1515import uuid
1616from copy import copy
1717from pathlib import Path
18- from typing import Any , Optional
18+ from typing import Any
1919
2020from gufe .storage .externalresource .base import ExternalStorage
2121
@@ -69,8 +69,13 @@ class Context:
6969
7070 Examples
7171 --------
72- >>> with Context(dag_label="my_dag", unit_label="unit1", scratch="/tmp/scratch",
73- ... shared_storage=shared_store, permanent_storage=perm_store) as ctx:
72+ >>> with Context(
73+ ... dag_label="my_dag",
74+ ... unit_label="unit1",
75+ ... scratch="/tmp/scratch",
76+ ... shared_storage=shared_store,
77+ ... permanent_storage=perm_store,
78+ ... ) as ctx:
7479 ... # use ctx within the ProtocolUnit._execute method
7580 ... pass
7681 """
@@ -82,8 +87,8 @@ def __init__(
8287 scratch : Path ,
8388 shared_storage : ExternalStorage ,
8489 permanent_storage : ExternalStorage ,
85- stderr : Optional [ Path ] = None ,
86- stdout : Optional [ Path ] = None ,
90+ stderr : Path | None = None ,
91+ stdout : Path | None = None ,
8792 ):
8893 self .scratch = scratch
8994 self .shared = StorageManager (
0 commit comments