diff --git a/CMakeLists.txt b/CMakeLists.txt index 9a96d0caf..78e778cf2 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -51,7 +51,7 @@ add_subdirectory(pybuda) ### Generate stubs for ttforge ### Run `cmake --build build -- make_stubs` to generate stubs add_custom_target(make_stubs - COMMAND pip install mypy + COMMAND pip install mypy==1.10 COMMAND stubgen -m pybuda._C -m pybuda._C.autograd -m pybuda._C.graph -m pybuda._C.torch_device -m pybuda._C.runtime -o pybuda -v COMMENT "Generating stubs for ttforge" USES_TERMINAL