From 03f6cf779d25687f88ba9f144be533d866a8b529 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Wed, 28 Aug 2024 12:48:29 -0700
Subject: [PATCH] Fix test setup

---
 libs/langgraph/Makefile       | 6 ++++--
 libs/langgraph/pyproject.toml | 1 -
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/libs/langgraph/Makefile b/libs/langgraph/Makefile
index feecd3fba..1500c84ef 100644
--- a/libs/langgraph/Makefile
+++ b/libs/langgraph/Makefile
@@ -20,16 +20,18 @@ start-postgres:
 stop-postgres:
 	docker compose -f tests/compose-postgres.yml down
 
+TEST_PATH ?= .
+
 test:
 	make start-postgres; \
-	poetry run pytest; \
+	poetry run pytest $(TEST_PATH); \
 	EXIT_CODE=$$?; \
 	make stop-postgres; \
 	exit $$EXIT_CODE
 
 test_watch:
 	make start-postgres; \
-	poetry run ptw .; \
+	poetry run ptw . -- --ff -v -x -n auto --dist worksteal --snapshot-update --tb short $(TEST_PATH); \
 	EXIT_CODE=$$?; \
 	make stop-postgres; \
 	exit $$EXIT_CODE
diff --git a/libs/langgraph/pyproject.toml b/libs/langgraph/pyproject.toml
index 8d1dfd694..de8d1fcd7 100644
--- a/libs/langgraph/pyproject.toml
+++ b/libs/langgraph/pyproject.toml
@@ -66,7 +66,6 @@ omit = ["tests/*"]
 [tool.pytest-watcher]
 now = true
 delay = 0.1
-runner_args = ["--ff", "-v", "-x", "-n", "auto", "--dist", "worksteal", "--snapshot-update", "--tb", "short"]
 patterns = ["*.py"]
 
 [build-system]