mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-10 07:28:15 -05:00
run unit test on windows ci (#9187)
* factor out testing_minimal in setup.py [pr] * testing_unit + windows
This commit is contained in:
26
setup.py
26
setup.py
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user