Skip to content

Commit e142e1c

Browse files
authored
fix import (#36409)
1 parent 6d30271 commit e142e1c

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tools/azure-sdk-tools/packaging_tools/multiapi_combiner.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -520,7 +520,11 @@ def serialize_operations_folder(self, async_mode: bool):
520520
):
521521
imports_to_add.append(i)
522522
elif i not in imports_splitlines:
523-
imports_to_add.append(i)
523+
error_name = i.strip(" ,\n")
524+
if error_name in ("StreamClosedError", "StreamConsumedError"):
525+
imports_to_add.append(f"from azure.core.exceptions import {error_name}")
526+
else:
527+
imports_to_add.append(i)
524528

525529
imports += "\n".join(imports_to_add)
526530
try:

0 commit comments

Comments
 (0)