#151 Support for functools.partial and flyte.map
Pull request opened by
kumare3
functools.maps can now be used with flyte.map.
Example
@env.task
def my_task_correct(batch_id: str, name: str, constant_param: str) -> str:
print(name, constant_param, batch_id)
return name
@env.task
def main():
compounds = [str(i) for i in range(3)]
constant_param = "shared_config"
task_with_constant_correct = partial(my_task_correct, constant_param=constant_param, name="daniel")
# This should work, as batch_id is the first parameter and the only one left for mapping
v = list(flyte.map(task_with_constant_correct, compounds, return_exceptions=False))
print("\n".join(v), flush=True)
flyteorg/flyte-sdk