[Projects] Add goals field (#608)

mlrun · Dec 20, 2020 · e5d3eb4 · e5d3eb4
1 parent de956e3
commit e5d3eb4
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/mlrun/api/schemas/project.py b/mlrun/api/schemas/project.py
@@ -20,6 +20,7 @@ class Config:
 
 class ProjectSpec(pydantic.BaseModel):
     description: typing.Optional[str] = None
+    goals: typing.Optional[str] = None
     params: typing.Optional[dict] = None
     functions: typing.Optional[list] = None
     workflows: typing.Optional[list] = None

diff --git a/mlrun/projects/project.py b/mlrun/projects/project.py
@@ -223,6 +223,7 @@ def __init__(
         source=None,
         subpath=None,
         origin_url=None,
+        goals=None,
     ):
         self.repo = None
 
@@ -233,6 +234,7 @@ def __init__(
         self.source = source or ""
         self.subpath = subpath or ""
         self.origin_url = origin_url or ""
+        self.goals = goals
         self.branch = None
         self.tag = ""
         self.params = params or {}

diff --git a/tests/api/api/test_projects.py b/tests/api/api/test_projects.py
@@ -12,7 +12,9 @@ def test_projects_crud(db: Session, client: TestClient) -> None:
     name1 = f"prj-{uuid4().hex}"
     project_1 = mlrun.api.schemas.Project(
         metadata=mlrun.api.schemas.ProjectMetadata(name=name1),
-        spec=mlrun.api.schemas.ProjectSpec(description="banana", source="source"),
+        spec=mlrun.api.schemas.ProjectSpec(
+            description="banana", source="source", goals="some goals"
+        ),
     )
 
     # create
@@ -39,7 +41,7 @@ def test_projects_crud(db: Session, client: TestClient) -> None:
     labels_2 = {"key": "value"}
     project_2 = mlrun.api.schemas.Project(
         metadata=mlrun.api.schemas.ProjectMetadata(name=name2, labels=labels_2),
-        spec=mlrun.api.schemas.ProjectSpec(description="banana", source="source"),
+        spec=mlrun.api.schemas.ProjectSpec(description="banana2", source="source2"),
     )
 
     # store

diff --git a/tests/rundb/test_httpdb.py b/tests/rundb/test_httpdb.py
@@ -528,6 +528,7 @@ def test_project_file_db_roundtrip(create_server):
 
     project_name = "project-name"
     description = "project description"
+    goals = "project goals"
     params = {"param_key": "param value"}
     artifact_path = "/tmp"
     conda = "conda"
@@ -546,6 +547,7 @@ def test_project_file_db_roundtrip(create_server):
         source=source,
         subpath=subpath,
         origin_url=origin_url,
+        goals=goals,
     )
     project = mlrun.projects.project.MlrunProject(
         metadata=project_metadata, spec=project_spec