added trl dependency

Signed-off-by: 1000850000 user <[email protected]>
foundation-model-stack · Sep 3, 2024 · 4831b35 · 4831b35
1 parent bc786c8
commit 4831b35
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
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"]
+dependencies = ["numba", "trl"]
 
 [tool.hatch.build.targets.wheel]
 only-include = ["src/fms_acceleration_aadp"]

diff --git a/plugins/attention-and-distributed-packing/tox.ini b/plugins/attention-and-distributed-packing/tox.ini
@@ -5,7 +5,6 @@ envlist = py, lint
 deps = 
     pytest>=7
     -e {toxinidir}
-skip_install = true
 commands = 
 
     # install the dependencies here to ensure