From 0039fb5a6fad3ad51b5c63c24af8d800aa9adfbd Mon Sep 17 00:00:00 2001
From: Kyunggeun Lee <quic_kyunggeu@quicinc.com>
Date: Thu, 19 Dec 2024 13:08:46 -0800
Subject: [PATCH] build: exclude unnecessary files from pip package (#3675)

Signed-off-by: Kyunggeun Lee <quic_kyunggeu@quicinc.com>
---
 TrainingExtensions/common/src/python/CMakeLists.txt | 2 ++
 TrainingExtensions/torch/src/python/CMakeLists.txt  | 1 +
 2 files changed, 3 insertions(+)

diff --git a/TrainingExtensions/common/src/python/CMakeLists.txt b/TrainingExtensions/common/src/python/CMakeLists.txt
index 683e0c34bd5..7f227ff38fb 100644
--- a/TrainingExtensions/common/src/python/CMakeLists.txt
+++ b/TrainingExtensions/common/src/python/CMakeLists.txt
@@ -49,5 +49,7 @@ else()
         DESTINATION ${CMAKE_INSTALL_LIBDIR}
         PATTERN "__pycache*" EXCLUDE
         PATTERN "CMakeLists.txt" EXCLUDE
+        PATTERN "_deps.pyi" EXCLUDE
+        PATTERN "_gen.py" EXCLUDE
         )
 endif()
diff --git a/TrainingExtensions/torch/src/python/CMakeLists.txt b/TrainingExtensions/torch/src/python/CMakeLists.txt
index 8a22e00e809..5201f4971a4 100644
--- a/TrainingExtensions/torch/src/python/CMakeLists.txt
+++ b/TrainingExtensions/torch/src/python/CMakeLists.txt
@@ -40,4 +40,5 @@ install(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
         DESTINATION ${CMAKE_INSTALL_LIBDIR}
         PATTERN "__pycache*" EXCLUDE
         PATTERN "CMakeLists.txt" EXCLUDE
+        PATTERN "examples" EXCLUDE
       )