feat: pass save results as parameter for node (#16)

entropy-lab · Jun 10, 2021 · cac24e9 · cac24e9
1 parent aa62b59
commit cac24e9
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 6 deletions.
diff --git a/entropylab_qpudb/_entropy_cal.py b/entropylab_qpudb/_entropy_cal.py
@@ -66,6 +66,7 @@ def __init__(
         dependency: Optional[Union[Node, Iterable[Node]]] = None,
         must_run_after: Set[Node] = None,
         name: Optional[str] = None,
+        save_results: bool = True,
     ):
         if dependency:
             if isinstance(dependency, Iterable):
@@ -111,7 +112,9 @@ def program(
 
         if name is None:
             name = self.__class__.__name__
-        super().__init__(name, program, input_vars, output_vars, must_run_after)
+        super().__init__(
+            name, program, input_vars, output_vars, must_run_after, save_results
+        )
 
     def add_config_dependency(self, node):
         super().add_input(f"config_{node.label}", node.outputs["config"])

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -23,7 +23,7 @@ homepage = "https://github.com/entropy-lab/entropy-qpu"
 python = "^3.7.1"
 ZODB = "^5.6.0"
 pandas = "^1.2.4"
-entropylab = "^0.1.1"
+entropylab = "^0.1.2"
 
 [tool.poetry.dev-dependencies]
 black = "^20.8b1"