[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

tests/examples/test_ips_lotf.py

@@ -1,13 +1,17 @@
 """Mock version of IPS LotF workflow for testing purposes."""
+
 import dataclasses
-import znflow
 import random
+
 import pytest
 
+import znflow
+
+
 @dataclasses.dataclass
 class AddData(znflow.Node):
     file: str
-    
+
     def run(self):
         if self.file is None:
             raise ValueError("File is None")
@@ -17,6 +21,7 @@ def run(self):
     def atoms(self):
         return "Atoms"
 
+
 @dataclasses.dataclass
 class TrainModel(znflow.Node):
     data: str
@@ -28,6 +33,7 @@ def run(self):
         self.model = "Model"
         print(f"Model: {self.model}")
 
+
 @dataclasses.dataclass
 class MD(znflow.Node):
     model: str
@@ -39,6 +45,7 @@ def run(self):
         self.atoms = "Atoms"
         print(f"Atoms: {self.atoms}")
 
+
 @dataclasses.dataclass
 class EvaluateModel(znflow.Node):
     model: str
@@ -52,10 +59,8 @@ def run(self):
         self.metrics = random.random()
         print(f"Metrics: {self.metrics}")
 
-@pytest.mark.parametrize(
-    "deployment",
-    ["vanilla_deployment", "dask_deployment"]
-)
+
+@pytest.mark.parametrize("deployment", ["vanilla_deployment", "dask_deployment"])
 def test_lotf(deployment, request):
     deployment = request.getfixturevalue(deployment)
 
@@ -72,6 +77,5 @@ def test_lotf(deployment, request):
             if znflow.resolve(metrics.metrics) == pytest.approx(0.623, 1e-3):
                 # break loop after 6th iteration
                 break
-
-    assert len(graph) == 22
 
+    assert len(graph) == 22

znflow/deployment/base.py

@@ -14,7 +14,7 @@ def run(self, nodes: t.Optional[t.List] = None):
         else:
             # convert nodes to UUIDs
             nodes = [node.uuid for node in nodes]
-        
+
         for node_uuid in nodes:
             node_available = self.graph.nodes[node_uuid].get("available", False)
             if self.graph.immutable_nodes and node_available:
@@ -27,4 +27,3 @@ def set_graph(self, graph: "DiGraph"):
     @abc.abstractmethod
     def _run_node(self, node_uuid):
         pass
-

znflow/deployment/dask_depl.py

@@ -64,21 +64,18 @@ def _run_node(self, node_uuid):
         predecessors = list(self.graph.predecessors(node_uuid))
         for predecessor in predecessors:
             self._run_node(predecessor)
-        
+
         node_available = self.graph.nodes[node_uuid].get("available", False)
         if self.graph.immutable_nodes and node_available:
             return
 
-
         self.results[node_uuid] = self.client.submit(
-                node_submit,
-                node=node,
-                predecessors={
-                    x: self.results[x] for x in self.results if x in predecessors
-                },
-                pure=False,
-                key=f"{node.__class__.__name__}-{node_uuid}",
-            )
+            node_submit,
+            node=node,
+            predecessors={x: self.results[x] for x in self.results if x in predecessors},
+            pure=False,
+            key=f"{node.__class__.__name__}-{node_uuid}",
+        )
         if self.graph.immutable_nodes:
             self.graph.nodes[node_uuid]["available"] = True
 

znflow/deployment/vanilla.py

@@ -1,6 +1,5 @@
 import dataclasses
 
-from networkx import predecessor
 
 from znflow import handler