Skip to content

Conversation

@arcticfly
Copy link
Contributor

No description provided.

Comment on lines 26 to 27
self._wandb_runs: dict[str, Run] = {}
self._weave_clients: dict[str, WeaveClient] = {}
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

i don't think we need this

Comment on lines 7 to 8
from wandb.sdk.wandb_run import Run
from weave.trace.weave_client import WeaveClient
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

i don't think we need this

Comment on lines 94 to 102
async def report_metrics(
self, *, model_id: str, step: int | Literal["latest"], metrics: dict[str, float]
) -> ReportMetricsResponse:
return await self._post(
f"/preview/models/{model_id}/checkpoints/{step}/report_metrics",
body={"metrics": metrics},
cast_to=ReportMetricsResponse,
options=dict(max_retries=0),
)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

i don't think we need this

Comment on lines 41 to 42
class ReportMetricsResponse(BaseModel):
success: bool
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

i don't think we need this

@arcticfly arcticfly changed the title [WIP]: Report serverless metrics Report serverless metrics Oct 3, 2025
@bradhilton bradhilton merged commit 523eb72 into main Oct 3, 2025
1 of 2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants