fixing diff cover location

qodo-ai · Jan 7, 2025 · 798ec11 · 798ec11
1 parent 0f13e8b
commit 798ec11
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Makefile b/Makefile
@@ -1,6 +1,6 @@
 # Makefile
 SITE_PACKAGES=$(shell python3 -c "import wandb, os; print(os.path.dirname(wandb.__file__))")
-DIFF_COVER_TEMPLATES=$(shell python -c "import site; print(site.getsitepackages()[0])")
+DIFF_COVER_TEMPLATES=$(shell python -c "import diff_cover, os; print(os.path.join(os.path.dirname(diff_cover.__file__), 'templates'))")
 TOML_FILES=$(shell find cover_agent/settings -name "*.toml" | sed 's/.*/-\-add-data "&:."/' | tr '\n' ' ')
 
 .PHONY: test build installer
@@ -24,7 +24,7 @@ installer:
 		$(TOML_FILES) \
 		--add-data "$(SITE_PACKAGES)/vendor:wandb/vendor" \
 		--add-data "build_helpers/anthropic_tokenizer.json:litellm/litellm_core_utils/tokenizers" \
-		--add-data "$(DIFF_COVER_TEMPLATES)/diff_cover/templates:diff_cover/templates" \
+		--add-data "$(DIFF_COVER_TEMPLATES):diff_cover/templates" \
 		--hidden-import=tiktoken_ext.openai_public \
 		--hidden-import=tiktoken_ext \
 		--hidden-import=wandb \
@@ -33,3 +33,4 @@ installer:
 		--onefile \
 		--name cover-agent \
 		cover_agent/main.py
+