chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,23 @@
|
||||
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]))
|
||||
Reference in New Issue
Block a user