wehub-resource-sync
20 lines
315 B
Python
20 lines
315 B
Python
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]))
|