Add support for testing experimental features (#4082)

albertvillanova · web-flow · commit 5139af371247 · 2025-09-16T07:46:48.000+02:00
diff --git a/Makefile b/Makefile
@@ -1,4 +1,4 @@
-.PHONY: test precommit common_tests slow_tests test_examples tests_gpu
+.PHONY: test precommit common_tests slow_tests test_examples tests_gpu test_experimental
 
 check_dirs := examples tests trl
 
@@ -28,3 +28,6 @@ test_examples:
 		TRL_ACCELERATE_CONFIG=$${file} bash $(COMMAND_FILES_PATH)/run_dpo.sh; \
 		echo $$?','$${file} >> temp_results_dpo_tests.txt; \
 	done
+
+test_experimental:
+	pytest -k "experimental"
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,3 +28,6 @@ markers = [
     "slow: marks tests as slow (deselect with '-m \"not slow\"')",
     "low-priority: marks tests as low priority (deselect with '-m \"not low-priority\"')"
 ]
+addopts = [
+    "-k", "not experimental",
+]

Original file line number	Diff line number	Diff line change
`@@ -28,3 +28,6 @@ markers = [`
`28`	`28`	`"slow: marks tests as slow (deselect with '-m \"not slow\"')",`
`29`	`29`	`"low-priority: marks tests as low priority (deselect with '-m \"not low-priority\"')"`
`30`	`30`	`]`
	`31`	`+addopts = [`
	`32`	`+ "-k", "not experimental",`
	`33`	`+]`