diff --git a/Makefile b/Makefile index b14f750..f2ed64e 100644 --- a/Makefile +++ b/Makefile @@ -16,12 +16,21 @@ build: pytest: . $(CONDA_BASE)/etc/profile.d/conda.sh && \ conda activate $(ENV_NAME) && \ - pytest test_models.py + pytest test_code.py pylint: . $(CONDA_BASE)/etc/profile.d/conda.sh && \ conda activate $(ENV_NAME) && \ pylint **/*.py +check: + . $(CONDA_BASE)/etc/profile.d/conda.sh && \ + conda activate $(ENV_NAME) && \ + pylint **/*.py && \ + pytest test_code.py + +run: + echo "The code is not completed yet." + clean: conda remove --name $(ENV_NAME) --all -y \ No newline at end of file diff --git a/README.md b/README.md index 32e1d7a..3c232f8 100644 --- a/README.md +++ b/README.md @@ -12,7 +12,7 @@

-Coding StableDiffusion from scratch using pytorch. +Coding Stable Diffusion from scratch using pytorch. ## Setup environment * Clone the repository and Go to StableDiffusion directory. @@ -24,8 +24,15 @@ git clone https://github.com/eljandoubi/StableDiffusion.git && cd StableDiffusio ```bash make build ``` - -## Clean environment. +## Check the code sanity +```bash +make check +``` +## Run the pipeline +```bash +make run +``` +## Clean environment ```bash make clean ``` \ No newline at end of file diff --git a/test_models.py b/test_code.py similarity index 100% rename from test_models.py rename to test_code.py