Skip to content

Commit 2da712c

Browse files
authored
teach stage_circuit to allow concurrency (#90)
1 parent 8aa0abc commit 2da712c

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/entitysdk/staging/circuit.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,9 @@
1212
L = logging.getLogger(__name__)
1313

1414

15-
def stage_circuit(client: Client, *, model: Circuit, output_dir: Path) -> Path:
15+
def stage_circuit(
16+
client: Client, *, model: Circuit, output_dir: Path, max_concurrent: int = 1
17+
) -> Path:
1618
"""Stage a Circuit directory into output_dir."""
1719
ensure_has_id(model)
1820
ensure_has_assets(model)
@@ -32,6 +34,7 @@ def stage_circuit(client: Client, *, model: Circuit, output_dir: Path) -> Path:
3234
asset_id=asset,
3335
output_path=output_dir,
3436
ignore_directory_name=True,
37+
max_concurrent=max_concurrent,
3538
)
3639

3740
L.debug("Downloaded circuit %s paths: %s", model.id, paths)

0 commit comments

Comments
 (0)