Skip to content

Commit 6f68f00

Browse files
[Task:4024] Fix breaking mypy checks
1 parent bbdc7e3 commit 6f68f00

File tree

5 files changed

+6
-6
lines changed

5 files changed

+6
-6
lines changed

src/zenml/integrations/mlflow/services/mlflow_deployment.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def run(self) -> None:
243243
# to run the deploy the model on the local running environment
244244
if int(mlflow_version[0]) >= 2:
245245
backend_kwargs["env_manager"] = "local"
246-
backend = PyFuncBackend( # type: ignore[no-untyped-call]
246+
backend = PyFuncBackend( # type: ignore[no-untyped-call, unused-ignore]
247247
config={},
248248
no_conda=True,
249249
workers=self.config.workers,
@@ -258,7 +258,7 @@ def run(self) -> None:
258258
"stack."
259259
)
260260
experiment_tracker.configure_mlflow()
261-
backend.serve( # type: ignore[no-untyped-call]
261+
backend.serve( # type: ignore[no-untyped-call, unused-ignore]
262262
model_uri=self.config.model_uri,
263263
port=self.endpoint.status.port,
264264
host="localhost",

src/zenml/integrations/mlflow/steps/mlflow_deployer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def mlflow_model_deployer_step(
114114
# Fetch the model URI from the MLflow artifact store
115115
model_uri = ""
116116
if mlflow_run_id and client.list_artifacts(mlflow_run_id, model_name):
117-
model_uri = artifact_utils.get_artifact_uri( # type: ignore[no-untyped-call]
117+
model_uri = artifact_utils.get_artifact_uri( # type: ignore[no-untyped-call, unused-ignore]
118118
run_id=mlflow_run_id, artifact_path=model_name
119119
)
120120

src/zenml/integrations/mlflow/steps/mlflow_registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def mlflow_register_model_step(
124124
if not model_source_uri and client.list_artifacts(
125125
mlflow_run_id, trained_model_name
126126
):
127-
model_source_uri = artifact_utils.get_artifact_uri( # type: ignore[no-untyped-call]
127+
model_source_uri = artifact_utils.get_artifact_uri( # type: ignore[no-untyped-call, unused-ignore]
128128
run_id=mlflow_run_id, artifact_path=trained_model_name
129129
)
130130
if not model_source_uri:

tests/harness/deployment/server_docker_mariadb.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ def up(self) -> None:
162162
Raises:
163163
RuntimeError: If the deployment could not be started.
164164
"""
165-
from compose.cli.main import ( # type: ignore[import-not-found]
165+
from compose.cli.main import ( # type: ignore
166166
TopLevelCommand,
167167
project_from_options,
168168
)

tests/harness/deployment/server_docker_mysql.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def up(self) -> None:
160160
Raises:
161161
RuntimeError: If the deployment could not be started.
162162
"""
163-
from compose.cli.main import ( # type: ignore[import-not-found]
163+
from compose.cli.main import ( # type: ignore
164164
TopLevelCommand,
165165
project_from_options,
166166
)

0 commit comments

Comments
 (0)