66from simvue .api .objects .artifact .base import ArtifactBase
77from simvue .api .objects .base import Sort
88from .file import FileArtifact
9+ from collections .abc import Generator
910from simvue .api .objects .artifact .object import ObjectArtifact
1011from simvue .api .request import get_json_from_response , get as sv_get
1112from simvue .api .url import URL
@@ -53,7 +54,7 @@ def from_run(
5354 run_id : str ,
5455 category : typing .Literal ["input" , "output" , "code" ] | None = None ,
5556 ** kwargs ,
56- ) -> typing . Generator [tuple [str , FileArtifact | ObjectArtifact ], None , None ]:
57+ ) -> Generator [tuple [str , FileArtifact | ObjectArtifact ]]:
5758 """Return artifacts associated with a given run.
5859
5960 Parameters
@@ -68,12 +69,12 @@ def from_run(
6869
6970 Returns
7071 -------
71- typing. Generator[tuple[str, FileArtifact | ObjectArtifact], None, None ]
72+ Generator[tuple[str, FileArtifact | ObjectArtifact]]
7273 The artifacts
7374
7475 Yields
7576 ------
76- Iterator[typing. Generator[tuple[str, FileArtifact | ObjectArtifact], None, None ]]
77+ Iterator[Generator[tuple[str, FileArtifact | ObjectArtifact]]]
7778 identifier for artifact
7879 the artifact itself as a class instance
7980
@@ -162,7 +163,7 @@ def get(
162163 offset : int | None = None ,
163164 sorting : list [ArtifactSort ] | None = None ,
164165 ** kwargs ,
165- ) -> typing . Generator [tuple [str , FileArtifact | ObjectArtifact ], None , None ]:
166+ ) -> Generator [tuple [str , FileArtifact | ObjectArtifact ]]:
166167 """Returns artifacts associated with the current user.
167168
168169 Parameters
0 commit comments