From b82781d2c2b82602c08ad666f28071524d1c97d2 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sat, 1 Feb 2025 10:08:42 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- tests/files/test_names.py | 9 ++++----- tests/unit_tests/test_node_name.py | 5 +++-- zntrack/project.py | 3 ++- 3 files changed, 9 insertions(+), 8 deletions(-) diff --git a/tests/files/test_names.py b/tests/files/test_names.py index a66a3b5a..1241894f 100644 --- a/tests/files/test_names.py +++ b/tests/files/test_names.py @@ -31,15 +31,14 @@ def test_names(proj_path): zntrack.examples.ParamsToMetrics(params={"loss": 0.09}, name="Lorine") # # zntrack.examples.ParamsToMetrics(params={"loss": 0.09}, name="Lorine") # ValueError - project.build() - assert json.loads( - (CWD / "zntrack_config" / "names.json").read_text() - ) == json.loads((proj_path / "zntrack.json").read_text()) + assert json.loads((CWD / "zntrack_config" / "names.json").read_text()) == json.loads( + (proj_path / "zntrack.json").read_text() + ) assert yaml.safe_load( (CWD / "dvc_config" / "names.yaml").read_text() ) == yaml.safe_load((proj_path / "dvc.yaml").read_text()) assert (CWD / "params_config" / "names.yaml").read_text() == ( proj_path / "params.yaml" - ).read_text() \ No newline at end of file + ).read_text() diff --git a/tests/unit_tests/test_node_name.py b/tests/unit_tests/test_node_name.py index b999bb70..c3960405 100644 --- a/tests/unit_tests/test_node_name.py +++ b/tests/unit_tests/test_node_name.py @@ -1,4 +1,5 @@ import pytest + import zntrack @@ -40,7 +41,7 @@ def test_duplicate_named_node_error(proj_path): """Test that an explicitly named node cannot be duplicated""" with zntrack.Project() as project: MyNode(name="A") - + with pytest.raises(ValueError, match="A node with the name 'A' already exists."): with project: MyNode(name="A") @@ -108,7 +109,7 @@ def test_nested_grouped_custom_node_names(proj_path): assert n1.name == "A_B_Alpha" assert n2.name == "A_B_Beta" - + project.build() assert n1.name == "A_B_Alpha" diff --git a/zntrack/project.py b/zntrack/project.py index 5b3ca13d..b37a8848 100644 --- a/zntrack/project.py +++ b/zntrack/project.py @@ -21,9 +21,10 @@ log = logging.getLogger(__name__) + class _FinalNodeNameString(str): """A string that represents the final name of a node. - + Used to differentiate between a custom name and a computed name. """