Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
@check_output(out_schema, "out_dataframe")
def test_func3(x, in_dataframe=None):
return {
"x": x,
"out_dataframe": in_dataframe.assign(f=["a", "b", "a"]),
}
@check_output(out_schema)
def test_func1(dataframe, x):
return dataframe.assign(f=["a", "b", "a"])
@check_output(out_schema)
def transform_secord_arg_with_dict_getter(self, x, df):
return _transform_helper(df)
@check_output(out_schema, 1)
def func_input_transform2(_, df):
return _, df
@check_output(out_schema)
def transform_first_arg(self, df):
return _transform_helper(df)
@check_output(out_schema)
def func_input_transform1(df):
return df
@check_output(out_schema, 0)
def test_func2(x, dataframe):
return dataframe.assign(f=["a", "b", "a"]), x
@check_output(out_schema)
def transform_secord_arg_with_dict_getter(self, x, df):
return _transform_helper(df)
@check_output(out_schema)
def transform_secord_arg_with_list_getter(self, x, df):
return _transform_helper(df)
@check_output(out_schema, 1)
def func_input_transform2(_, df):
return _, df