Skip to content

Commit

Permalink
add docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
lzh222333 authored and you-n-g committed Jun 30, 2021
1 parent 0b83fb3 commit b523b27
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions qlib/workflow/recorder.py
Original file line number Diff line number Diff line change
Expand Up @@ -309,6 +309,18 @@ def save_objects(self, local_path=None, artifact_path=None, **kwargs):
shutil.rmtree(temp_dir)

def load_object(self, name):
"""
Load object such as prediction file or model checkpoint in mlflow.
Args:
name (str): the object name
Raises:
LoadObjectError: if raise some exceptions when load the object
Returns:
object: the saved object in mlflow.
"""
assert self.uri is not None, "Please start the experiment and recorder first before using recorder directly."

try:
Expand Down

0 comments on commit b523b27

Please sign in to comment.