ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-06-26 23:15:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add support for Dataset projects

Browse Source
This commit is contained in:
allegroai
2022-07-08 17:45:03 +03:00
parent ee87778548
commit 3a60f00d93
8 changed files with 114 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
apiserver/tests/automated/test_subprojects.py
View File

@@ -12,6 +12,19 @@ from apiserver.tests.automated import TestService
class TestSubProjects(TestService):
def test_dataset_stats(self):
project = self._temp_project(name="Dataset test", system_tags=["dataset"])
res = self.api.organization.get_entities_count(datasets={"system_tags": ["dataset"]})
self.assertEqual(res.datasets, 1)
task = self._temp_task(project=project)
data = self.api.projects.get_all_ex(id=[project], include_dataset_stats=True).projects[0]
self.assertIsNone(data.dataset_stats)
self.api.tasks.edit(task=task, runtime={"ds_file_count": 2, "ds_total_size": 1000})
data = self.api.projects.get_all_ex(id=[project], include_dataset_stats=True).projects[0]
self.assertEqual(data.dataset_stats, {"file_count": 2, "total_size": 1000})
def test_project_aggregations(self):
"""This test requires user with user_auth_only... credentials in db"""
user2_client = APIClient(