FilesExpand file tree

 master

/

hatch_build.py

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

43 lines (31 loc) · 1.5 KB

 master

/

hatch_build.py

Top

File metadata and controls

• Code
• Blame

43 lines (31 loc) · 1.5 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

from __future__ import absolute_import

import os

import sys

from hatchling.metadata.plugin.interface import MetadataHookInterface

class CustomMetadataHook(MetadataHookInterface):

def update(self, metadata):

metadata["optional-dependencies"] = get_optional_dependencies(self.root)

def get_optional_dependencies(root):

def read_feature_deps(feature):

req_file = os.path.join(root, "requirements", "extras", f"{feature}_requirements.txt")

with open(req_file, encoding="utf-8") as f:

return list(filter(lambda d: not d.startswith("#"), f.read().splitlines()))

optional_dependencies = {"all": []}

for feature in ("feature-processor", "huggingface", "local", "scipy", "sagemaker-mlflow"):

dependencies = read_feature_deps(feature)

optional_dependencies[feature] = dependencies

optional_dependencies["all"].extend(dependencies)

# Test dependencies come last because we don't want them in `all`

optional_dependencies["test"] = read_feature_deps("test")

optional_dependencies["test"].extend(optional_dependencies["all"])

# remove torch and torchvision if python version is not 3.10/3.11

if sys.version_info.minor not in (10, 11):

optional_dependencies["test"] = list(

filter(

lambda d: not d.startswith(

("sentencepiece", "transformers", "torch", "torchvision")

),

optional_dependencies["test"],

)

)

return optional_dependencies