Skip to content

Commit

Permalink
More mypy
Browse files Browse the repository at this point in the history
  • Loading branch information
schustmi committed Nov 7, 2024
1 parent 92522d3 commit 929f5a0
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ def svc_trainer(
) -> Tuple[
# This third argument marks this as a Model Artifact
Annotated[ClassifierMixin, ArtifactConfig("trained_model", is_model_artifact=True)],
# This third argument marks this as a Data Artifact
# This third argument marks this as a service Artifact
Annotated[str, ArtifactConfig("deployment_uri", is_deployment_artifact=True)],
]:
...
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
"""

from collections import defaultdict
from typing import Dict, Set
from typing import Any, Dict, Set

import sqlalchemy as sa
from alembic import op
Expand Down Expand Up @@ -99,7 +99,7 @@ def resolve_duplicate_entities() -> None:
needs_new_name,
needs_new_number,
) in needs_update:
values = {}
values: Dict[str, Any] = {}

is_numeric_version = str(number) == name
next_numeric_version = max(existing_numbers[model_id]) + 1
Expand Down

0 comments on commit 929f5a0

Please sign in to comment.