ran pre-commits

Author: zilto

hamilton/graph.py

@@ -733,7 +733,12 @@ def from_modules(
         """
 
         functions = sum([find_functions(module) for module in modules], [])
-        return FunctionGraph.from_functions(*functions, config=config, adapter=adapter, allow_module_overrides=allow_module_overrides)
+        return FunctionGraph.from_functions(
+            *functions,
+            config=config,
+            adapter=adapter,
+            allow_module_overrides=allow_module_overrides,
+        )
 
     @staticmethod
     def from_functions(
@@ -743,7 +748,10 @@ def from_functions(
         allow_module_overrides: bool = False,
     ) -> "FunctionGraph":
         nodes = create_function_graph(
-            *functions, config=config, adapter=adapter, allow_module_overrides=allow_module_overrides
+            *functions,
+            config=config,
+            adapter=adapter,
+            allow_module_overrides=allow_module_overrides,
         )
         return FunctionGraph(nodes, config, adapter)
 

hamilton/graph_utils.py

@@ -25,7 +25,7 @@ def find_functions(function_module: ModuleType) -> List[Tuple[str, Callable]]:
     :return: list of tuples of (func_name, function).
     """
     return [
-        (name, fn) for name, fn in inspect.getmembers(function_module)
-        if is_hamilton_function(fn)
-        and is_submodule(inspect.getmodule(fn), function_module)
+        (name, fn)
+        for name, fn in inspect.getmembers(function_module)
+        if is_hamilton_function(fn) and is_submodule(inspect.getmodule(fn), function_module)
     ]

tests/execution/test_node_grouping.py

@@ -98,7 +98,9 @@ def bar(foo: int) -> int:
     # This is hacking around function graph which is messy as it is built of larger components
     # (modules), and should instead be broken into smaller pieces (functions/nodes), and have utilities
     # to create it from those.
-    fn_graph = graph.FunctionGraph.from_modules(ad_hoc_utils.create_temporary_module(bar), config={})
+    fn_graph = graph.FunctionGraph.from_modules(
+        ad_hoc_utils.create_temporary_module(bar), config={}
+    )
     node_ = fn_graph.nodes["bar"]
     task = grouping.TaskSpec(
         base_id="bar",
@@ -120,7 +122,9 @@ def bar(foo: int, baz: int = 1) -> int:
     # This is hacking around function graph which is messy as it is built of larger components
     # (modules), and should instead be broken into smaller pieces (functions/nodes), and have utilities
     # to create it from those.
-    fn_graph = graph.FunctionGraph.from_modules(ad_hoc_utils.create_temporary_module(bar), config={})
+    fn_graph = graph.FunctionGraph.from_modules(
+        ad_hoc_utils.create_temporary_module(bar), config={}
+    )
     node_ = fn_graph.nodes["bar"]
     task = grouping.TaskSpec(
         base_id="bar",

tests/function_modifiers/test_adapters.py

@@ -570,7 +570,9 @@ def test_loader_default_factory_field():
     def foo(param: int) -> int:
         return param
 
-    fn_graph = graph.FunctionGraph.from_modules(ad_hoc_utils.create_temporary_module(foo), config={})
+    fn_graph = graph.FunctionGraph.from_modules(
+        ad_hoc_utils.create_temporary_module(foo), config={}
+    )
     assert len(fn_graph.nodes) == 3
     assert "foo" in fn_graph.nodes
 
@@ -599,7 +601,9 @@ def test_saver_default_factory_field():
     def foo(param: int) -> int:
         return param
 
-    fn_graph = graph.FunctionGraph.from_modules(ad_hoc_utils.create_temporary_module(foo), config={})
+    fn_graph = graph.FunctionGraph.from_modules(
+        ad_hoc_utils.create_temporary_module(foo), config={}
+    )
     assert len(fn_graph.nodes) == 3
     assert "foo" in fn_graph.nodes