Deployment schemas (NA-513) (#84)

* deployment of pipeline schemas * Refactor schemes for future metrics changes (#74) * Refactor schemes for future metrics changes * Update version * Add schemas for hardware related metrics (#75) * Add schemas for hardware metrics * Add project_id to metrics * add a few tests for pipeline cloud (#72) * add a few tests for pipeline cloud and have to cleanup too many flake8 errs * Refactor schemes for future metrics changes (#74) * Refactor schemes for future metrics changes * Update version * Add schemas for hardware related metrics (#75) * Add schemas for hardware metrics * Add project_id to metrics * Refactor schemes for future metrics changes (#74) * Refactor schemes for future metrics changes * Update version * Add schemas for hardware related metrics (#75) * Add schemas for hardware metrics * Add project_id to metrics * add a few tests for pipeline cloud and have to cleanup too many flake8 errs Co-authored-by: andrei-trandafir <47391556+andrei-trandafir@users.noreply.github.com> * add same checks made pre-commit to CI workflow (#76) * add same checks made pre-commit to CI workflow * wrongly sorted file was alredy in codebase * just checking how to set path * rename old job to lint again * adding flake8 error to see job catch it * removed flake8 error * fixing small details on dev setup script (#73) * now we good bruh * smol changes to script * poetry run pytest so no need to activate env * permissions * added empty line back to readme * add compute type field to run create schema (#77) * add compute type field to run create schema * pydantic enum serialization fix * move rsc type to runGet schema * add comment * v bump * Add created_at and updated_at fields to Token schema. Towards NA-247. * Add expires_at to TokenCreate. Towards NA-247. * Bump version to 0.0.27. * Schema change * Deployment schemas update * Class name change * Deployment schema change * Update to schema * Additional schema fields in deployment * Additional schema fields in deployment * import order reformat * Version bump * RM PipelineGet Project * Import cleanup * VB Co-authored-by: andrei-trandafir <47391556+andrei-trandafir@users.noreply.github.com> Co-authored-by: Max Schulkin <maxemijo@gmail.com> Co-authored-by: Alex Pearce <alex@mystic.ai>
mystic-ai · Apr 25, 2022 · f28e887 · f28e887
1 parent bf6c0fb
commit f28e887
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 7 deletions.
diff --git a/pipeline/schemas/deployment.py b/pipeline/schemas/deployment.py
@@ -0,0 +1,21 @@
+from typing import Optional
+
+from pipeline.schemas.base import BaseModel
+from pipeline.schemas.pipeline import PipelineGet
+from pipeline.schemas.project import ProjectGet
+
+
+class DeploymentCreate(BaseModel):
+    project_id: str
+    pipeline_id: str
+
+
+class DeploymentGet(BaseModel):
+    id: str
+    pipeline: PipelineGet
+    project: ProjectGet
+    active: bool
+
+
+class DeploymentPatch(BaseModel):
+    active: Optional[bool]
diff --git a/pipeline/schemas/pipeline.py b/pipeline/schemas/pipeline.py
@@ -72,3 +72,4 @@ class PipelineCreate(BaseModel):
     models: List[ModelGet]
     graph_nodes: List[PipelineGraphNode]
     outputs: List[str]
+    project_id: Optional[str]
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,16 +1,19 @@
 [tool.poetry]
 name = "pipeline-ai"
-version = "0.0.31"
+version = "0.0.32"
 description = "Pipelines for machine learning workloads."
-authors = ["Paul Hetherington <ph@mystic.ai>", "Alex Pearwin <alex@mystic.ai>", "Maximiliano Schulkin <max@mystic.ai>", "Neil Wang <neil@mystic.ai>"]
-packages = [
-    {include="pipeline"}
+authors = [
+  "Paul Hetherington <ph@mystic.ai>",
+  "Alex Pearwin <alex@mystic.ai>",
+  "Maximiliano Schulkin <max@mystic.ai>",
+  "Neil Wang <neil@mystic.ai>",
 ]
+packages = [{ include = "pipeline" }]
 readme = "README.md"
 license = "Apache-2.0"
 classifiers = [
-    'License :: OSI Approved :: Apache Software License',
-    'Programming Language :: Python :: 3',
+  'License :: OSI Approved :: Apache Software License',
+  'Programming Language :: Python :: 3',
 ]
 
 
@@ -23,7 +26,7 @@ setuptools = "^59.2.0"
 pyhumps = "^3.0.2"
 requests-toolbelt = "^0.9.1"
 tqdm = "^4.62.3"
-PyYAML = {version = "^6.0", optional = true}
+PyYAML = { version = "^6.0", optional = true }
 
 [tool.poetry.dev-dependencies]
 setuptools = "^59.2.0"