ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 17:17:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Local modules (except trains) imported from a folder inside the git project should not be logged as "local packages", they should be ignored

Browse Source
This commit is contained in:
allegroai 2020-04-09 12:21:37 +03:00
parent 07daf8f5e6
commit ffedb219d5
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -38,9 +38,9 @@ class GenerateReqs(object):
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])
ourselves = self.__module__.split('.')[0]if self.__module__ else None
if ourselves and ourselves not in candidates:
candidates.add(ourselves)
logger.info('Check module in local environment.')
for name in candidates:
@ -54,6 +54,11 @@ class GenerateReqs(object):
# add local modules, so we know what is used but not installed.
for name in self._local_mods:
if name in modules:
if ourselves and name == ourselves:
from ...version import __version__
reqs.add(name, __version__, modules[name])
continue
# if this is a folder of our project, we can safely ignore it
if os.path.commonpath([os.path.realpath(self._project_path)]) == \
os.path.commonpath([os.path.realpath(self._project_path),