Skip to content

Commit

Permalink
Merge branch 'ft-1397-resolving-misusage-no-models-ptandloggenerator'…
Browse files Browse the repository at this point in the history
… into 'integration'

FT 1397 Resolving misusage of no_models in ptandloggenerator

See merge request process-mining/pm4py/pm4py-core!533
  • Loading branch information
fit-sebastiaan-van-zelst committed Dec 14, 2021
2 parents 597f7af + f547854 commit 51181d6
Showing 1 changed file with 14 additions and 2 deletions.
16 changes: 14 additions & 2 deletions pm4py/algo/simulation/tree_generator/variants/ptandloggenerator.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,13 +107,25 @@ def apply(parameters: Optional[Dict[Union[str, Parameters], Any]] = None) -> obj
if not "infrequent" in parameters:
parameters["infrequent"] = 0.5
if not "no_models" in parameters:
parameters["no_models"] = 10
parameters["no_models"] = 1
if not "unfold" in parameters:
parameters["unfold"] = 10
if not "max_repeat" in parameters:
parameters["max_repeat"] = 10

return GeneratedTree(parameters).generate()
no_models = parameters["no_models"]

if no_models == 1:
return GeneratedTree(parameters).generate()
else:
# if the generation of an higher number of models is required,
# proceed to the generation of these and return a list of
# process trees.
ret = []
for i in range(no_models):
ret.append(GeneratedTree(parameters).generate())
return ret


def assign_operator(operator):
if operator == "choice":
Expand Down

0 comments on commit 51181d6

Please sign in to comment.