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

Add TRAINS_SUPPRESS_UPDATE_MESSAGE #157

Browse Source
This commit is contained in:
allegroai 2020-07-02 01:24:47 +03:00
parent 5cc09d83fd
commit 59b78b288e
3 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
trains/backend_interface/task/task.py
View File

@ -41,7 +41,7 @@ from ...storage.helper import StorageHelper, StorageError
from .access import AccessMixin from .access import AccessMixin
from .log import TaskHandler from .log import TaskHandler
from .repo import ScriptInfo from .repo import ScriptInfo
from ...config import config, PROC_MASTER_ID_ENV_VAR from ...config import config, PROC_MASTER_ID_ENV_VAR, SUPPRESS_UPDATE_MESSAGE_ENV_VAR
class Task(IdObjectBase, AccessMixin, SetupUploadMixin): class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
@ -235,7 +235,7 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
# check latest version # check latest version
from ...utilities.check_updates import CheckPackageUpdates from ...utilities.check_updates import CheckPackageUpdates
latest_version = CheckPackageUpdates.check_new_package_available(only_once=True) latest_version = CheckPackageUpdates.check_new_package_available(only_once=True)
if latest_version: if latest_version and not SUPPRESS_UPDATE_MESSAGE_ENV_VAR.get(default=config.get('development.suppress_update_message', False)):
if not latest_version[1]: if not latest_version[1]:
sep = os.linesep sep = os.linesep
self.get_logger().report_text( self.get_logger().report_text(
@ -526,15 +526,15 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
self.reload() self.reload()
def started(self, ignore_errors=True): def started(self, ignore_errors=True, force=False):
# type: (bool) -> () # type: (bool, bool) -> ()
""" The signal that this Task started. """ """ The signal that this Task started. """
return self.send(tasks.StartedRequest(self.id), ignore_errors=ignore_errors) return self.send(tasks.StartedRequest(self.id, force=force), ignore_errors=ignore_errors)
def stopped(self, ignore_errors=True): def stopped(self, ignore_errors=True, force=False):
# type: (bool) -> () # type: (bool, bool) -> ()
""" The signal that this Task stopped. """ """ The signal that this Task stopped. """
return self.send(tasks.StoppedRequest(self.id), ignore_errors=ignore_errors) return self.send(tasks.StoppedRequest(self.id, force=force), ignore_errors=ignore_errors)
def completed(self, ignore_errors=True): def completed(self, ignore_errors=True):
# type: (bool) -> () # type: (bool) -> ()

4
trains/config/default/sdk.conf
View File

@ -147,6 +147,10 @@
# do not analyze the entire repository. # do not analyze the entire repository.
force_analyze_entire_repo: false force_analyze_entire_repo: false
# If set to true, *trains* update message will not be printed to the console
# this value can be overwritten with os environment variable TRAINS_SUPPRESS_UPDATE_MESSAGE=1
suppress_update_message: false
# Development mode worker # Development mode worker
worker { worker {
# Status report period in seconds # Status report period in seconds

2
trains/config/defs.py
View File

@ -20,6 +20,8 @@ TRAINS_CACHE_DIR = EnvEntry("TRAINS_CACHE_DIR", "ALG_CACHE_DIR")
LOG_LEVEL_ENV_VAR = EnvEntry("TRAINS_LOG_LEVEL", "ALG_LOG_LEVEL", converter=or_(int, str)) LOG_LEVEL_ENV_VAR = EnvEntry("TRAINS_LOG_LEVEL", "ALG_LOG_LEVEL", converter=or_(int, str))
SUPPRESS_UPDATE_MESSAGE_ENV_VAR = EnvEntry("TRAINS_SUPPRESS_UPDATE_MESSAGE", "ALG_SUPPRESS_UPDATE_MESSAGE", type=bool)
# Repository detection # Repository detection
VCS_REPO_TYPE = EnvEntry("TRAINS_VCS_REPO_TYPE", "ALG_VCS_REPO_TYPE", default="git") VCS_REPO_TYPE = EnvEntry("TRAINS_VCS_REPO_TYPE", "ALG_VCS_REPO_TYPE", default="git")
VCS_REPOSITORY_URL = EnvEntry("TRAINS_VCS_REPO_URL", "ALG_VCS_REPO_URL") VCS_REPOSITORY_URL = EnvEntry("TRAINS_VCS_REPO_URL", "ALG_VCS_REPO_URL")