ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 21:33:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix requirements detection to make sure trains is detected even if we execute without actually being installed

Browse Source
This commit is contained in:
allegroai 2020-04-01 19:04:57 +03:00
parent 01772430d6
commit 41ca1a2e49
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
trains/utilities/pigar/__main__.py
View File

@ -33,20 +33,22 @@ class GenerateReqs(object):
self._project_path, self._ignores) self._project_path, self._ignores)
if module_callback: if module_callback:
modules = module_callback(modules) modules = module_callback(modules)
app_name = os.path.basename(self._project_path)
if app_name in local_mods:
local_mods.remove(app_name)
# Filtering modules # Filtering modules
candidates = self._filter_modules(modules, local_mods) candidates = self._filter_modules(modules, local_mods)
# make sure we are in candidates
ourselves = self.__module__.split('.') if self.__module__ else None
if ourselves and ourselves[0] not in candidates:
candidates.add(ourselves[0])
logger.info('Check module in local environment.') logger.info('Check module in local environment.')
for name in candidates: for name in candidates:
logger.info('Checking module: %s', name) logger.info('Checking module: %s', name)
if name in self._installed_pkgs: if name in self._installed_pkgs:
pkg_name, version = self._installed_pkgs[name] pkg_name, version = self._installed_pkgs[name]
reqs.add(pkg_name, version, modules[name]) reqs.add(pkg_name, version, modules[name])
else: elif name in modules:
guess.add(name, 0, modules[name]) guess.add(name, 0, modules[name])
# add local modules, so we know what is used but not installed. # add local modules, so we know what is used but not installed.