Ignore files within a .git directory

* Do not consider files within a .git directory
when we are using separate-weights

Author: 8W9aG

pkg/weights/weights.go

@@ -30,6 +30,9 @@ func FindWeights(fw FileWalker) ([]string, []string, error) {
 		if info.IsDir() {
 			return nil
 		}
+		if isGitFile(path) {
+			return nil
+		}
 		if isCodeFile(path) {
 			codeFiles = append(codeFiles, path)
 			return nil
@@ -97,6 +100,10 @@ func isCodeFile(path string) bool {
 	return ext == ".py" || ext == ".ipynb"
 }
 
+func isGitFile(path string) bool {
+	return strings.Contains(path, ".git")
+}
+
 // filterDirsContainingCode filters out directories that contain code files.
 // If a dir is a prefix for any given codeFiles, it will be filtered out.
 func filterDirsContainingCode(dirs []string, codeFiles []string) []string {

pkg/weights/weights_test.go

@@ -160,3 +160,19 @@ func TestSubDirMerge(t *testing.T) {
 	require.Empty(t, rootFiles)
 	require.Equal(t, []string{"models"}, dirs)
 }
+
+// Test case for ignoring files within a .git directory
+func TestIgnoreGitFiles(t *testing.T) {
+	mockFileWalker := func(root string, walkFn filepath.WalkFunc) error {
+		sizes := []int64{sizeThreshold, sizeThreshold, 1024}
+		for i, path := range []string{".git/root-large", "root-large", "predict.py"} {
+			walkFn(path, mockFileInfo{size: sizes[i]}, nil)
+		}
+		return nil
+	}
+
+	dirs, rootFiles, err := FindWeights(mockFileWalker)
+	require.NoError(t, err)
+	require.Equal(t, []string{"root-large"}, rootFiles)
+	require.Empty(t, dirs)
+}