ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-03-16 00:07:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Preinstall numpy if it exists in the requirements (temporary fix)

Browse Source
This commit is contained in:
allegroai 2020-02-04 18:06:25 +02:00
parent 6d091d8e08
commit 28d752d568
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
trains_agent/helper/package/cython_req.py
View File

@ -6,14 +6,14 @@ from .requirements import SimpleSubstitution
class CythonRequirement(SimpleSubstitution): class CythonRequirement(SimpleSubstitution):
name = "cython" name = ("cython", "numpy", )
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super(CythonRequirement, self).__init__(*args, **kwargs) super(CythonRequirement, self).__init__(*args, **kwargs)
def match(self, req): def match(self, req):
# match both Cython & cython # match both Cython & cython
return req.name and self.name == req.name.lower() return req.name and req.name.lower() in self.name
def replace(self, req): def replace(self, req):
""" """