Skip to content

Commit 7b80746

Browse files
authored
Create _dynamic_deps.py
1 parent ad372b4 commit 7b80746

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

_dynamic_deps.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
"""
2+
Dynamic dependencies for ethnicolr.
3+
This module determines platform-specific dependencies.
4+
"""
5+
6+
import platform
7+
8+
def get_tensorflow_requirement():
9+
"""Get the appropriate TensorFlow requirement based on platform."""
10+
if platform.machine() == 'aarch64':
11+
return "tensorflow-aarch64>=2.7.2,<2.16"
12+
else:
13+
return "tensorflow>=2.7.2,<2.16"
14+
15+
def get_dependencies():
16+
"""Return the full list of dependencies including platform-specific ones."""
17+
deps = [
18+
get_tensorflow_requirement(),
19+
"pandas>=1.3.0",
20+
"numpy>=1.20.0",
21+
"h5py>=3.1.0",
22+
"tqdm>=4.62.0"
23+
]
24+
return deps

0 commit comments

Comments
 (0)