Makefile

.PHONY: help setup train evaluate inference precommit format clean

.DEFAULT_GOAL := help

help:
	@echo "VardaGPT - Memory-enhanced GPT-2 model powered by Hugging Face Transformers and FAISS"
	@echo "Usage:"
	@echo "  make <command>"
	@echo ""
	@echo "Commands:"
	@echo "  help        Display this help message"
	@echo "  setup       Set up the project by creating a virtual environment and installing dependencies"
	@echo "  train       Train the VardaGPT model"
	@echo "  evaluate    Evaluate the trained model on validation and testing sets"
	@echo "  inference   Generate text using the memory-enhanced GPT-2 model"
	@echo "  precommit   Run pre-commit hooks manually on all files"
	@echo "  format      Format code using black, flake8, mypy, and prettier"
	@echo "  clean       Clean up the project directory by removing virtual environment and temporary files"

setup:
	python -m venv venv
	source venv/bin/activate
	pip install -r requirements.txt

train:
	source venv/bin/activate
	python src/train.py

train-parallel:
	source venv/bin/activate
	python src/train_parallel.py

evaluate:
	source venv/bin/activate
	python src/evaluate.py

inference:
	source venv/bin/activate
	python src/inference.py --prompt "Your prompt text here"

precommit:
	pre-commit run --all-files

format:
	pre-commit run --all-files

test:
	coverage run -m pytest --log-cli-level=DEBUG --capture=tee-sys -v .

clean:
	rm -rf venv/
	find . -type f -name "*.pyc" -exec rm -f {} \;
	find . -type d -name "__pycache__" -exec rm -rf {} \;