-
Notifications
You must be signed in to change notification settings - Fork 11
/
meta.yaml
81 lines (71 loc) · 1.82 KB
/
meta.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
# changed the recipe to be client, server at the same time
# add pytorch (or other NN libraries, see bioimageio.core) to enable server
# capabilities.
# Add git, inferno, pytorch!
{% set setup_py_data = load_setup_py_data() %}
package:
name: tiktorch
version: {{ setup_py_data['version'] }}
source:
path: ../
build:
noarch: python
number: 0
script:
- python -m pip install . --no-deps --ignore-installed --no-cache-dir -vvv
entry_points:
- tiktorch-server = tiktorch.server.base:main
requirements:
host:
- pip
- python >=3.8
- setuptools >=40.0
- setuptools_scm
run:
- python >=3.8
{% for dep in setup_py_data['install_requires'] %}
- {{ dep.lower() }}
{% endfor %}
run_constrained:
- mkl <2024.1.0 # [linux] until pytorch is compatible with the current version
- cudatoolkit >=10.2
{% for dep in setup_py_data['extras_require']['server-pytorch'] %}
- {{ dep.lower() }}
{% endfor %}
about:
home: https://github.com/ilastik/tiktorch
license: MIT
license_family: MIT
license_file: LICENSE
summary: 'Neural network server'
doc_url: https://github.com/ilastik/tiktorch
dev_url: https://github.com/ilastik/tiktorch
test:
requires:
{% for dep in setup_py_data['extras_require']['server-pytorch'] %}
- {{ dep.lower() }}
{% endfor %}
- python 3.9.*
- pytest
- pytest-grpc
- pytest-timeout
- cpuonly
- ilastik-pytorch-version-helper-cpu
imports:
# client
- tiktorch
- tiktorch.proto.inference_pb2
- tiktorch.proto.inference_pb2_grpc
- tiktorch.proto.data_store_pb2
- tiktorch.proto.data_store_pb2_grpc
- tiktorch.types
- tiktorch.converters
# server
- tiktorch.server.base
- tiktorch.server.session
source_files:
- tests
- tiktorch
- pytest.ini
commands:
- pytest