-
Notifications
You must be signed in to change notification settings - Fork 14
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Co-authored-by: Abhinav Chobey <[email protected]>
- Loading branch information
1 parent
3e4399d
commit ea23124
Showing
1 changed file
with
21 additions
and
25 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -741,7 +741,7 @@ def update_dataset_url(self, artifact: mlpb.Artifact, updated_url: str): | |
``` | ||
Args: | ||
artifact: Artifact for which url is to be updated | ||
updated_url: The updated url path of the dataset. | ||
updated_url: The updated url path of the dataset. | ||
Returns: | ||
Updates artifact in mlmd, does not returns anything. | ||
""" | ||
|
@@ -1780,22 +1780,18 @@ def metadata_push(pipeline_name,filename,execution_id: str = ""): | |
return output | ||
|
||
def metadata_pull(pipeline_name,filename ="./mlmd", execution_id: str = ""): | ||
""" Pulls MLMD file from CMF-server. | ||
""" Pulls MLMD file from CMF-server. | ||
Example: | ||
```python | ||
result = metadata_pull("example_pipeline", "./mlmd_directory", "execution_123") | ||
``` | ||
Args: | ||
pipeline_name: Name of the pipeline. | ||
filename: File path to store the MLMD file. | ||
execution_id: Optional execution ID. | ||
Returns: | ||
Message from the _metadata_pull function. | ||
""" | ||
|
||
|
||
# Required arguments: pipeline_name, filename(file path to store mlmd file) | ||
#Optional arguments: Execution_ID | ||
output = _metadata_pull(pipeline_name,filename, execution_id) | ||
|
@@ -1889,31 +1885,31 @@ def cmf_init(type: str="", | |
```python | ||
cmf_init( type="local", | ||
path="/path/to/re", | ||
git_remote_url="[email protected]:user/repo.git", | ||
git_remote_url="[email protected]:user/repo.git", | ||
cmf_server_url="http://cmf-server" | ||
neo4j_user", | ||
neo4j_password="password", | ||
neo4j_uri="bolt://localhost:76" | ||
neo4j_uri="bolt://localhost:76" | ||
) | ||
``` | ||
Args: | ||
type: Type of repository ("local", "minioS3", "amazonS3", "sshremote") | ||
path: Path for the local repository. | ||
git_remote_url: Git remote URL for version control. | ||
cmf_server_url: CMF server URL. | ||
neo4j_user: Neo4j database username. | ||
neo4j_password: Neo4j database password. | ||
neo4j_uri: Neo4j database URI. | ||
url: URL for MinioS3 or AmazonS3. | ||
Args: | ||
type: Type of repository ("local", "minioS3", "amazonS3", "sshremote") | ||
path: Path for the local repository. | ||
git_remote_url: Git remote URL for version control. | ||
cmf_server_url: CMF server URL. | ||
neo4j_user: Neo4j database username. | ||
neo4j_password: Neo4j database password. | ||
neo4j_uri: Neo4j database URI. | ||
url: URL for MinioS3 or AmazonS3. | ||
endpoint_url: Endpoint URL for MinioS3. | ||
access_key_id: Access key ID for MinioS3 or AmazonS3. | ||
secret_key: Secret key for MinioS3 or AmazonS3. | ||
user: SSH remote username. | ||
password: SSH remote password. | ||
port: SSH remote port | ||
Returns: | ||
Output based on the initialized repository type. | ||
. """ | ||
access_key_id: Access key ID for MinioS3 or AmazonS3. | ||
secret_key: Secret key for MinioS3 or AmazonS3. | ||
user: SSH remote username. | ||
password: SSH remote password. | ||
port: SSH remote port | ||
Returns: | ||
Output based on the initialized repository type. | ||
""" | ||
|
||
if type=="": | ||
return print("Error: Type is not provided") | ||
|