Skip input_schema for hftransformersv2 back compat (#88)

* Skip input_schema for hftransformersv2 back compat

Signed-off-by: Walter Martin <[email protected]>

* lint

Signed-off-by: Walter Martin <[email protected]>

* only check for hftransformers if input arg is dict

Signed-off-by: Walter Martin <[email protected]>

---------

Signed-off-by: Walter Martin <[email protected]>

Author: wamartin-aml

inference_schema/schema_decorators.py

@@ -43,9 +43,13 @@ def input_schema(param_name, param_type, convert_to_provided_type=True, optional
 
     @_schema_decorator(attr_name=INPUT_SCHEMA_ATTR, schema=swagger_schema, supported_versions=supported_versions)
     def decorator_input(user_run, instance, args, kwargs):
-        if convert_to_provided_type:
+        is_hftransformersv2 = False
+        if len(args) > 0 and type(args[0]) is dict:
+            args_keys = args[0].keys()
+            is_hftransformersv2 = len(args_keys) == 2 and "parameters" in args_keys and "input_string" in args_keys
+        # skip all of this for hftransformersv2
+        if convert_to_provided_type and not is_hftransformersv2:
             args = list(args)
-
             if param_name not in kwargs.keys() and not optional:
                 decorators = _get_decorators(user_run)
                 arg_names = inspect.getfullargspec(decorators[-1]).args

tests/conftest.py

@@ -266,6 +266,19 @@ def standard_py_func(param):
     return standard_py_func
 
 
+@pytest.fixture(scope="session")
+def decorated_standard_func_parameters(standard_sample_input, sample_param_dict):
+    @input_schema('input_data', StandardPythonParameterType(standard_sample_input))
+    @input_schema('params', StandardPythonParameterType(sample_param_dict), optional=False)
+    def standard_params_func(input_data, params=None):
+        if params is not None:
+            assert type(params) is dict
+        beams = params['num_beams'] if params is not None else 0
+        return input_data["input_string"], beams
+
+    return standard_params_func
+
+
 @pytest.fixture(scope="session")
 def standard_sample_input_multitype_list():
     return ['foo', 1]

tests/test_standard_parameter_type.py

@@ -52,3 +52,15 @@ def test_float_int_handling(self, decorated_float_func):
         int_input = 1
         result = decorated_float_func(int_input)
         assert int_input == result
+
+    def test_standard_params_handling_hftransformersv2(self, decorated_standard_func_parameters):
+        input_data = {
+            "input_string": ["the meaning of life is"],
+            "parameters": {
+                "num_beams": 2,
+                "max_length": 512
+            }
+        }
+        result = decorated_standard_func_parameters(input_data)
+        assert result[0][0] == "the meaning of life is"
+        assert result[1] == 0