From 2d91d4cde65bb254d70f6d2959e4f44f670509fd Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Tue, 14 Jan 2020 11:37:41 +0200
Subject: [PATCH] Add support for "-e ./folder" lines in requirements

---
 trains_agent/helper/package/cython_req.py   | 2 +-
 trains_agent/helper/package/horovod_req.py  | 2 +-
 trains_agent/helper/package/requirements.py | 4 ++--
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/trains_agent/helper/package/cython_req.py b/trains_agent/helper/package/cython_req.py
index 8f803cc..947e28e 100644
--- a/trains_agent/helper/package/cython_req.py
+++ b/trains_agent/helper/package/cython_req.py
@@ -13,7 +13,7 @@ class CythonRequirement(SimpleSubstitution):
 
     def match(self, req):
         # match both Cython & cython
-        return self.name == req.name.lower()
+        return req.name and self.name == req.name.lower()
 
     def replace(self, req):
         """
diff --git a/trains_agent/helper/package/horovod_req.py b/trains_agent/helper/package/horovod_req.py
index 6aa8c6d..ff1be9d 100644
--- a/trains_agent/helper/package/horovod_req.py
+++ b/trains_agent/helper/package/horovod_req.py
@@ -14,7 +14,7 @@ class HorovodRequirement(SimpleSubstitution):
 
     def match(self, req):
         # match both horovod
-        return self.name == req.name.lower()
+        return req.name and self.name == req.name.lower()
 
     def post_install(self):
         if self.post_install_req:
diff --git a/trains_agent/helper/package/requirements.py b/trains_agent/helper/package/requirements.py
index 5157c0d..20ba404 100644
--- a/trains_agent/helper/package/requirements.py
+++ b/trains_agent/helper/package/requirements.py
@@ -48,7 +48,7 @@ class MarkerRequirement(object):
 
     def tostr(self, markers=True):
         if not self.uri:
-            parts = [self.name]
+            parts = [self.name or self.line]
 
             if self.extras:
                 parts.append('[{0}]'.format(','.join(sorted(self.extras))))
@@ -258,7 +258,7 @@ class RequirementsManager(object):
                 warning('could not resolve python wheel replacement for {}'.format(req))
                 raise
             except Exception:
-                warning('could not resolve python wheel replacement for {}, '
+                warning('could not resolve python wheel replacement for \"{}\", '
                         'using original requirements line: {}'.format(req, i))
                 return None