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

Fix worker Id is used instead of worker key when processing report

Browse Source
This commit is contained in:
allegroai 2022-07-08 17:48:17 +03:00
parent c18eb99d06
commit c28f478ea8
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apiserver/bll/workers/__init__.py
View File

@ -143,7 +143,7 @@ class WorkerBLL:
self._log_stats_to_es( self._log_stats_to_es(
company_id=company_id, company_id=company_id,
company_name=entry.company.name, company_name=entry.company.name,
worker=report.worker, worker=entry.key,
timestamp=report.timestamp, timestamp=report.timestamp,
task=report.task, task=report.task,
machine_stats=report.machine_stats, machine_stats=report.machine_stats,

10
apiserver/tests/automated/test_workers.py
View File

@ -79,7 +79,8 @@ class TestWorkersService(TestService):
self.api.workers.status_report(**data) self.api.workers.status_report(**data)
time.sleep(1) time.sleep(1)
return workers res = self.api.workers.get_all(last_seen=100)
return [w.key for w in res.workers]
def _create_running_task(self, task_name): def _create_running_task(self, task_name):
task_input = dict( task_input = dict(
@ -109,7 +110,8 @@ class TestWorkersService(TestService):
def test_get_stats(self): def test_get_stats(self):
workers = self._simulate_workers() workers = self._simulate_workers()
to_date = utc_now_tz_aware()
to_date = utc_now_tz_aware() + timedelta(seconds=10)
from_date = to_date - timedelta(days=1) from_date = to_date - timedelta(days=1)
# no variants # no variants
@ -195,8 +197,8 @@ class TestWorkersService(TestService):
self._simulate_workers() self._simulate_workers()
to_date = utc_now_tz_aware() to_date = utc_now_tz_aware() + timedelta(seconds=10)
from_date = to_date - timedelta(minutes=10) from_date = to_date - timedelta(minutes=1)
# no variants # no variants
res = self.api.workers.get_activity_report( res = self.api.workers.get_activity_report(