Skip to content

Commit

Permalink
remove unused pydantic validation class
Browse files Browse the repository at this point in the history
  • Loading branch information
Galileo-Galilei committed Nov 22, 2024
1 parent 6fc30d4 commit cb7e9f5
Showing 1 changed file with 1 addition and 8 deletions.
9 changes: 1 addition & 8 deletions kedro_mlflow/mlflow/kedro_pipeline_model.py
Original file line number Diff line number Diff line change
@@ -1,24 +1,17 @@
import logging
from pathlib import Path
from typing import Any, Dict, Optional, Union
from typing import Dict, Optional, Union

from kedro.framework.hooks import _create_hook_manager
from kedro.io import DataCatalog, MemoryDataset
from kedro.pipeline import Pipeline
from kedro.runner import AbstractRunner, SequentialRunner
from kedro_datasets.pickle import PickleDataset
from mlflow.pyfunc import PythonModel
from pydantic import BaseModel

from kedro_mlflow.pipeline.pipeline_ml import PipelineML


class PredictParamsSchema(BaseModel):
parameters: Optional[dict[str, Any]] = {}
# runner: AbstractRunner
# hooks: Iterable[Any] # cf. _register_hooks


class KedroPipelineModel(PythonModel):
def __init__(
self,
Expand Down

0 comments on commit cb7e9f5

Please sign in to comment.