From 62a21a3072addd3d6951bedff024679d32547f63 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Mon, 18 Nov 2024 22:43:23 -0800
Subject: [PATCH] add comment for the noqa

---
 docs/openapi_generator/generate.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/docs/openapi_generator/generate.py b/docs/openapi_generator/generate.py
index b055746ad..46bc32297 100644
--- a/docs/openapi_generator/generate.py
+++ b/docs/openapi_generator/generate.py
@@ -31,7 +31,12 @@ from .strong_typing.schema import json_schema_type
 
 schema_utils.json_schema_type = json_schema_type
 
-from llama_stack.distribution.stack import LLAMA_STACK_API_VERSION, LlamaStack  # noqa
+# this line needs to be here to ensure json_schema_type has been altered before
+# the imports use the annotation
+from llama_stack.distribution.stack import (  # noqa: E402
+    LLAMA_STACK_API_VERSION,
+    LlamaStack,
+)
 
 
 def main(output_dir: str):