Skip to content

Commit

Permalink
Alignment status (#65)
Browse files Browse the repository at this point in the history
* Alignment status

* Linter
  • Loading branch information
tleyden authored Jul 15, 2024
1 parent 8294122 commit bf866a9
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 1 deletion.
2 changes: 2 additions & 0 deletions arcee/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@

from arcee import config
from arcee.api import (
alignment_status,
corpus_status,
delete_corpus,
deployment_status,
Expand Down Expand Up @@ -57,4 +58,5 @@
"list_pretrainings",
"deployment_status",
"merging_status",
"alignment_status",
]
14 changes: 13 additions & 1 deletion arcee/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -313,7 +313,7 @@ def merging_status(merging: str) -> Dict[str, str]:
Check the status of a merging job
Args:
merging (str): The name of the deployment to check the status
merging (str): The name of the alignment to check the status
"""

data = {"merging_name": merging}
Expand Down Expand Up @@ -389,6 +389,18 @@ def start_alignment(
return make_request("post", Route.alignment + "/startAlignment", data)


def alignment_status(alignment: str) -> Dict[str, str]:
"""
Check the status of an alignment job
Args:
alignment (str): The name of the alignment to check the status
"""

data = {"alignment_name": alignment}

return make_request("get", Route.alignment + "/status", data)


def upload_alignment(alignment_name: str, alignment_id: str, qa_set_id: str, pretraining_id: str) -> Dict[str, str]:
data = {
"alignment_name": alignment_name,
Expand Down

0 comments on commit bf866a9

Please sign in to comment.