fix: add version boundaries as per fms-hf-tuning. Latest does not seem to have the collator

kmehant · kmehant · commit af08d694c6b6 · 2025-09-06T19:02:03.000+05:30
Signed-off-by: Mehant Kammakomati &lt;mehant.kammakomati2@ibm.com&gt;
diff --git a/plugins/attention-and-distributed-packing/pyproject.toml b/plugins/attention-and-distributed-packing/pyproject.toml
@@ -23,7 +23,7 @@ classifiers=[
     "Programming Language :: Python :: 3.11",
 ]
 
-dependencies = ["numba", "trl"]
+dependencies = ["numba", "trl>=0.19.1,<0.20.0"]
 
 [tool.hatch.build.targets.wheel]
 only-include = ["src/fms_acceleration_aadp"]

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@ classifiers=[`
`23`	`23`	`"Programming Language :: Python :: 3.11",`
`24`	`24`	`]`
`25`	`25`
`26`		`-dependencies = ["numba", "trl"]`
	`26`	`+dependencies = ["numba", "trl>=0.19.1,<0.20.0"]`
`27`	`27`
`28`	`28`	`[tool.hatch.build.targets.wheel]`
`29`	`29`	`only-include = ["src/fms_acceleration_aadp"]`