ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-04 02:57:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix detect installed packages when Trains is not imported directly.

Browse Source
This commit is contained in:
allegroai 2020-07-06 21:00:01 +03:00
parent 6221c6c096
commit 2b75c933ee
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -77,6 +77,8 @@ class GenerateReqs(object):
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]
if name not in modules:
modules.add(name, name, 0)
reqs.add(pkg_name, version, modules[name]) reqs.add(pkg_name, version, modules[name])
elif name in modules: elif name in modules:
guess.add(name, 0, modules[name]) guess.add(name, 0, modules[name])