run unit test on windows ci (#9187)

* factor out testing_minimal in setup.py [pr]

* testing_unit + windows
This commit is contained in:
chenyu
2025-02-20 14:40:41 -05:00
committed by GitHub
parent 287de4ecc6
commit 3e22747799
3 changed files with 21 additions and 15 deletions

View File

@@ -7,6 +7,14 @@ directory = Path(__file__).resolve().parent
with open(directory / 'README.md', encoding='utf-8') as f:
long_description = f.read()
testing_minimal = [
"numpy",
"torch",
"pytest",
"pytest-xdist",
"hypothesis",
]
setup(name='tinygrad',
version='0.10.1',
description='You like pytorch? You like micrograd? You love tinygrad! <3',
@@ -35,19 +43,14 @@ setup(name='tinygrad',
"types-tqdm",
],
#'mlperf': ["mlperf-logging @ git+https://github.com/mlperf/logging.git@4.1.0-rc3"],
'testing_minimal': [
"numpy",
"torch",
"pytest",
"pytest-xdist",
"hypothesis",
'testing_minimal': testing_minimal,
'testing_unit': testing_minimal + [
"tqdm",
"safetensors",
"tabulate" # for sz.py
],
'testing': [
"numpy",
"torch",
'testing': testing_minimal + [
"pillow",
"pytest",
"pytest-xdist",
"onnx==1.16.0",
"onnx2torch",
"opencv-python",
@@ -60,7 +63,6 @@ setup(name='tinygrad',
"blobfile",
"librosa",
"networkx",
"hypothesis",
"nibabel",
"bottle",
"ggml-python",