(internal): setup flamegraph generation for python generator (#4196)

fern-api · Aug 3, 2024 · 3684271 · 3684271
1 parent 7ec3428
commit 3684271
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/generators/python/sdk/Dockerfile b/generators/python/sdk/Dockerfile
@@ -31,3 +31,11 @@ COPY ./src ./src
 RUN poetry install
 
 ENTRYPOINT ["python", "-m", "src.fern_python.generators.sdk.cli"]
+
+## NOTE: Uncomment the below to generate a flame graph for the python generator. 
+##       To visualize the flamegraph you can run: 
+##         - poetry add snakeviz
+##         - poetry run snakeviz output.prof
+## 
+## RUN mkdir -p /fern/output
+## ENTRYPOINT ["python", "-m", "cProfile", "-o", "/fern/output/output.prof", "/src/fern_python/generators/sdk/cli.py"]