Fixes inconsistency between pre-commit and git hook

So the git commit hook for isort was running with
different configuration from the pre-commit CI run.

This (a) fixes files to be consistent with (b) the configuration
now provided -- e.g. what is first party, and what is local.

This should hopefully now prevent any conflicts from reoccurring.

Author: skrawcz

.pre-commit-config.yaml

@@ -25,7 +25,7 @@ repos:
     rev: '5.12.0'
     hooks:
     -   id: isort
-        args: ["--profile", "black", --line-length=100]
+        args: ["--profile", "black", "--line-length=100", "--known-local-folder", "tests", "-p", "hamilton"]
 -   repo: https://github.com/pycqa/flake8
     rev: 6.1.0
     hooks:

contrib/setup.cfg

@@ -15,3 +15,4 @@ ignore =
 
 [isort]
 known_first_party=hamilton
+known_local_folder=tests

setup.cfg

@@ -15,3 +15,4 @@ ignore =
 
 [isort]
 known_first_party=hamilton
+known_local_folder=tests

tests/execution/test_executors.py

@@ -3,14 +3,6 @@
 
 import numpy as np
 import pytest
-from tests.resources.dynamic_parallelism import (
-    inputs_in_collect,
-    no_parallel,
-    parallel_collect_multiple_arguments,
-    parallel_complex,
-    parallel_delayed,
-    parallel_linear_basic,
-)
 
 import hamilton.ad_hoc_utils
 from hamilton import base, driver
@@ -30,6 +22,15 @@
 )
 from hamilton.htypes import Collect, Parallelizable
 
+from tests.resources.dynamic_parallelism import (
+    inputs_in_collect,
+    no_parallel,
+    parallel_collect_multiple_arguments,
+    parallel_complex,
+    parallel_delayed,
+    parallel_linear_basic,
+)
+
 ADAPTER = base.DefaultAdapter()
 
 

tests/execution/test_node_grouping.py

@@ -1,5 +1,3 @@
-from tests.resources.dynamic_parallelism import no_parallel, parallel_complex, parallel_linear_basic
-
 from hamilton import ad_hoc_utils, base, graph, node
 from hamilton.execution import grouping
 from hamilton.execution.grouping import (
@@ -12,6 +10,8 @@
 from hamilton.graph import FunctionGraph
 from hamilton.node import NodeType
 
+from tests.resources.dynamic_parallelism import no_parallel, parallel_complex, parallel_linear_basic
+
 
 def test_group_individually():
     fn_graph = FunctionGraph.from_modules(no_parallel, config={})

tests/function_modifiers/test_macros.py

@@ -3,7 +3,6 @@
 
 import pandas as pd
 import pytest
-import tests.resources.pipe
 
 import hamilton.function_modifiers
 from hamilton import base, driver, function_modifiers, models, node
@@ -12,6 +11,8 @@
 from hamilton.function_modifiers.macros import Applicable, ensure_function_empty, pipe, step
 from hamilton.node import DependencyType
 
+import tests.resources.pipe
+
 
 def test_no_code_validator():
     def no_code():

tests/function_modifiers/test_recursive.py

@@ -3,7 +3,6 @@
 from typing import Tuple
 
 import pytest
-import tests.resources.reuse_subdag
 
 from hamilton import ad_hoc_utils, graph
 from hamilton.function_modifiers import (
@@ -18,6 +17,8 @@
 from hamilton.function_modifiers.dependencies import source
 from hamilton.function_modifiers.recursive import _validate_config_inputs
 
+import tests.resources.reuse_subdag
+
 
 def test_collect_function_fns():
     val = random.randint(0, 100000)

tests/function_modifiers/test_validation.py

@@ -1,11 +1,6 @@
 import numpy as np
 import pandas as pd
 import pytest
-from tests.resources.dq_dummy_examples import (
-    DUMMY_VALIDATORS_FOR_TESTING,
-    SampleDataValidator2,
-    SampleDataValidator3,
-)
 
 from hamilton import node
 from hamilton.data_quality.base import DataValidationError, ValidationResult
@@ -17,6 +12,12 @@
 )
 from hamilton.node import DependencyType
 
+from tests.resources.dq_dummy_examples import (
+    DUMMY_VALIDATORS_FOR_TESTING,
+    SampleDataValidator2,
+    SampleDataValidator3,
+)
+
 
 def test_check_output_node_transform():
     decorator = check_output(

tests/io/test_materialization.py

@@ -2,8 +2,6 @@
 from typing import Any, Collection, Dict, List, Optional, Tuple, Type
 
 import pytest
-import tests.resources.cyclic_functions
-import tests.resources.test_default_args
 
 from hamilton import base, graph, node, registry
 from hamilton.function_modifiers import load_from, save_to, value
@@ -20,6 +18,9 @@
     to,
 )
 
+import tests.resources.cyclic_functions
+import tests.resources.test_default_args
+
 global_mock_data_saver_cache = {}
 
 

tests/test_caching.py

@@ -3,12 +3,13 @@
 
 import pandas as pd
 import pytest
-from tests import nodes
 
 from hamilton import base
 from hamilton.driver import Driver
 from hamilton.experimental import h_cache
 
+from tests import nodes
+
 
 @pytest.fixture
 def df():