Fixed code registry but need to remove duplicated json encoding code from hf runtime

Author: RobertSamoilescu

mlserver/codecs/json.py

@@ -12,7 +12,7 @@
 from .string import decode_str
 from .lists import as_list
 from .utils import InputOrOutput, SingleInputRequestCodec
-from .base import InputCodec, register_input_codec, register_request_codec
+from .base import InputCodec
 from ..types import RequestInput, ResponseOutput, Parameters
 
 
@@ -95,7 +95,6 @@ def _is_nested_primitives(obj):
     return False
 
 
-@register_input_codec
 class JSONCodec(InputCodec):
     """
     Encodes a list of Python objects as a BYTES input (output).
@@ -160,7 +159,6 @@ def encode_input(
         )
 
 
-@register_request_codec
 class JSONRequestCodec(SingleInputRequestCodec):
     """
     Decodes the first input (output) of request (response) as a NumPy array.

runtimes/mlflow/mlserver_mlflow/runtime.py

@@ -22,6 +22,7 @@
 from mlserver.errors import InferenceError
 from mlserver.settings import ModelParameters
 from mlserver.logging import logger
+from mlserver.codecs import JSONCodec
 
 from .codecs import TensorDictCodec
 from .metadata import (
@@ -195,7 +196,7 @@ def _sync_metadata(self) -> None:
         )
 
     async def predict(self, payload: InferenceRequest) -> InferenceResponse:
-        decoded_payload = self.decode_request(payload)
+        decoded_payload = self.decode_request(payload, default_codec=JSONCodec)
         params = None
         if payload.parameters and payload.parameters.model_extra:
             params = payload.parameters.model_extra