튜토리얼: 레지스트리 아티팩트 별칭 자동화

이 튜토리얼에서는 아티팩트 메타데이터로 트리거되는 레지스트리 자동화를 구축하는 방법을 안내합니다. 레지스트리의 아티팩트에 특정 별칭(예: production)이 지정되면 W&B가 웹훅으로 POST 요청을 보냅니다. 모델을 정해진 단계로 승격할 때마다 배포 파이프라인, 페이징 서비스, 알림 채널 등의 다운스트림 시스템에 알리려면 이 패턴을 사용하세요. 이 튜토리얼은 레지스트리에서 모델 라이프사이클을 관리하는 ML 엔지니어와 MLOps 실무자를 대상으로 합니다.

프로젝트 자동화를 만드는 방법은 튜토리얼: 프로젝트 run 실패 알림 자동화를 참조하세요.

사전 요구 사항

Team Settings에서 설정한 웹훅
컬렉션이 하나 이상 있는 W&B 레지스트리. 기존 레지스트리를 재사용해도 됩니다.

레지스트리 자동화 생성

레지스트리 범위의 자동화를 설정하면, 레지스트리 내 모든 컬렉션에서 아티팩트에 특정 별칭(예: production)이 지정될 때마다 W&B가 웹훅으로 POST 요청을 보냅니다.

레지스트리를 열고 Automations 탭을 클릭한 다음 Create automation을 클릭하세요.
이벤트로 An artifact alias is added를 선택하세요. 대상 별칭(예: production 또는 staging)과 일치하는 Alias regex를 입력하세요.
Next step을 클릭하세요. Action type을 Webhooks로 설정하고 사용할 웹훅을 선택하세요. 웹훅에 페이로드가 필요하면 JSON 본문을 붙여 넣고 ${artifact_collection_name}, ${artifact_version_string} 같은 페이로드 변수를 사용하세요.
Next step을 클릭하세요. 자동화 이름과 설명(선택)을 입력한 다음 Create automation을 클릭하세요.

자세한 내용은 웹훅 자동화 만들기(Registry 탭)를 참조하세요.

자동화 테스트

자동화가 처음부터 끝까지 제대로 실행되는지 확인하려면 아티팩트 버전에 별칭을 추가하여 설정한 이벤트를 트리거하세요. W&B App 또는 공개 API를 사용하여 레지스트리의 아티팩트 버전에 별칭(예: production)을 추가하면 됩니다. 예를 들면 다음과 같습니다.

import wandb

with wandb.init(project="my-project") as run:
    artifact = wandb.Artifact("my-model", type="model")
    # ... 필요에 따라 아티팩트에 파일이나 메타데이터를 로깅합니다 ...
    run.log_artifact(artifact)
    run.wait()  # 다음 단계로 넘어가기 전에 아티팩트 로깅이 완료될 때까지 기다립니다

# 컬렉션의 최신 버전에 별칭을 추가합니다
api = wandb.Api()
collection = api.artifact_collection(name="my-model", type_name="model")
version = next(collection.versions())  # 최신 버전을 조회합니다

version.aliases.append("production")
version.save()
print("Added alias 'production' to", version.name)

잠시 후 웹훅 엔드포인트에 설정한 페이로드가 담긴 POST 요청이 전송됩니다. 이제 이 레지스트리의 아티팩트에 조건과 일치하는 별칭이 적용될 때마다 실행되는 레지스트리 자동화가 완성되었습니다.

더 알아보기

W&B 자동화에 대해 이 튜토리얼에서 다루지 않은 내용을 더 알아보려면 다음 리소스를 참조하세요:

프로젝트 및 레지스트리의 모든 이벤트 유형에 대한 Automation events and scopes.
전체 UI 및 페이로드 세부 정보에 대한 Slack 자동화 만들기 및 웹훅 자동화 만들기.

​사전 요구 사항

​레지스트리 자동화 생성

​자동화 테스트

​더 알아보기

사전 요구 사항

레지스트리 자동화 생성

자동화 테스트

더 알아보기