chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,19 @@
|
||||
import sys
|
||||
|
||||
import mlflow
|
||||
|
||||
|
||||
def call_tracking_apis():
|
||||
mlflow.log_metric("some_key", 3)
|
||||
|
||||
|
||||
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