This commit is contained in:
allegroai 2021-01-24 09:12:12 +02:00
parent 1988cf8adf
commit dd03ca0d1d
2 changed files with 4 additions and 1 deletions

View File

@ -42,7 +42,8 @@ class ScriptRequirements(object):
reqs, try_imports, guess, local_pks = gr.extract_reqs( reqs, try_imports, guess, local_pks = gr.extract_reqs(
module_callback=ScriptRequirements.add_trains_used_packages, entry_point_filename=entry_point_filename) module_callback=ScriptRequirements.add_trains_used_packages, entry_point_filename=entry_point_filename)
return self.create_requirements_txt(reqs, local_pks) return self.create_requirements_txt(reqs, local_pks)
except Exception: except Exception as ex:
_logger.warning("Failed auto-generating package requirements: {}".format(ex))
return '', '' return '', ''
@staticmethod @staticmethod

View File

@ -426,6 +426,8 @@ def _search_path(path):
dev_dir = line dev_dir = line
if not dev_dir: if not dev_dir:
continue continue
if not os.path.exists(dev_dir):
continue
# Egg info path. # Egg info path.
info_dir = [_file for _file in os.listdir(dev_dir) info_dir = [_file for _file in os.listdir(dev_dir)
if _file.endswith('egg-info')] if _file.endswith('egg-info')]