From e96d50f4f03b4bfbceb1480a2a03c99c2653736a Mon Sep 17 00:00:00 2001 From: Jess Frazelle Date: Wed, 6 Apr 2022 23:18:49 -0700 Subject: [PATCH] update Signed-off-by: Jess Frazelle --- generate/generate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/generate/generate.py b/generate/generate.py index 934ace1b8..467ed85cb 100755 --- a/generate/generate.py +++ b/generate/generate.py @@ -164,7 +164,7 @@ response: Response[""" + success_type + """] = await """ + fn_name + """.asyncio } # Add our imports. - f.write("from typing import Any, Dict, Optional, Union\n") + f.write("from typing import Any, Dict, Optional, Union, cast\n") f.write("\n") f.write("import httpx\n") f.write("\n") @@ -669,7 +669,7 @@ def generateType(path: str, name: str, schema: dict): has_date_time = hasDateTime(schema) if has_date_time: f.write("import datetime\n") - f.write("from typing import Any, Dict, List, Type, TypeVar, Union\n") + f.write("from typing import Any, Dict, List, Type, TypeVar, Union, cast\n") f.write("\n") f.write("import attr\n") if has_date_time: