Update unit tests for CI (#952)

* Update CI

* Revert submodule pointer update

* Update

* update

* update
This commit is contained in:
Sebastian Raschka
2026-01-27 17:44:55 -06:00
committed by GitHub
parent 59d9262047
commit e155d1b02c
5 changed files with 86 additions and 9 deletions

View File

@@ -14,12 +14,10 @@ dependencies = [
"torch>=2.2.2; sys_platform == 'darwin' and platform_machine == 'arm64' and python_version <= '3.12'",
"torch>=2.2.2; sys_platform == 'linux' and python_version <= '3.12'",
"torch>=2.2.2; sys_platform == 'win32' and python_version <= '3.12'",
"tensorflow>=2.16.2; sys_platform == 'darwin' and platform_machine == 'x86_64'",
"tensorflow>=2.18.0; sys_platform == 'darwin' and platform_machine == 'arm64'",
"tensorflow>=2.18.0; sys_platform == 'linux'",
"tensorflow>=2.18.0; sys_platform == 'win32'",
"jupyterlab>=4.0",
"tiktoken>=0.5.1",
"matplotlib>=3.7.1",
@@ -53,7 +51,7 @@ bonus = [
"sentencepiece>=0.1.99",
"thop",
"tokenizers>=0.21.1",
"transformers>=4.33.2",
"transformers>=5.0.0",
"tqdm>=4.65.0",
]