Skip to content
Snippets Groups Projects
Commit 1f5b452b authored by vlorentz's avatar vlorentz Committed by Phabricator Migration user
Browse files

luigi: Add an option to define the maximum RAM used by graph compression

parent 92a8d059
No related branches found
No related tags found
No related merge requests found
......@@ -93,6 +93,7 @@ class CompressGraph(luigi.Task):
Larger is faster, but consumes more resources.
""",
)
max_ram = luigi.Parameter(default=None, significant=False)
object_types = list(ObjectType)
# To make this configurable, we could use this:
......@@ -142,6 +143,8 @@ class CompressGraph(luigi.Task):
if self.batch_size:
conf["batch_size"] = self.batch_size
if self.max_ram:
conf["max_ram"] = self.max_ram
# Delete stamps. Otherwise interrupting this compression pipeline may leave
# stamps from a previous successful compression
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment