我正在使用一个临时文件,然后将我的数据转储到json,然后将该数据作为工件记录到GCP中。临时文件名称是随机生成的,因此有没有办法将其重命名为我选择的自定义名称,以便具有正确命名约定的工件被MLFlow记录?
代码示例:
with mlflow.start_run():
with tempfile.NamedTemporaryFile(mode='w', suffix='.json', delete=False) as f:
original_path = f.name
new_filename = 'metadata_' + pipeline_run_id
json.dump(user_args, f)
f.name = new_filename
f.seek(0)
mlflow.log_artifact(f.name)
os.unlink(original_path)
assert not os.path.exists(original_path)