File tree Expand file tree Collapse file tree 5 files changed +6
-6
lines changed
src/zenml/integrations/mlflow Expand file tree Collapse file tree 5 files changed +6
-6
lines changed Original file line number Diff line number Diff 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" ,
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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 :
Original file line number Diff line number Diff 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 )
Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments