wehub-resource-sync
24 lines
477 B
Python
24 lines
477 B
Python
import sys
|
|
import tempfile
|
|
|
|
import mlflow
|
|
|
|
|
|
def call_tracking_apis():
|
|
mlflow.log_metric("some_key", 3)
|
|
with tempfile.NamedTemporaryFile("w") as temp_file:
|
|
temp_file.write("Temporary content.")
|
|
mlflow.log_artifact(temp_file.name)
|
|
|
|
|
|
def main(use_start_run):
|
|
if use_start_run:
|
|
with mlflow.start_run():
|
|
call_tracking_apis()
|
|
else:
|
|
call_tracking_apis()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(use_start_run=int(sys.argv[1]))
|