#917 - Integrated the PrimaiteSession into all tests.

- Ran a full pre-commit hook and thus encountered tons of fixes required
2023-06-30 09:08:13 +01:00
parent 7f912df383
commit 73015802ec
62 changed files with 1880 additions and 802 deletions
--- a/tests/test_training_config.py
+++ b/tests/test_training_config.py
@@ -16,7 +16,9 @@ def test_legacy_lay_down_config_yaml_conversion():
    with open(new_path, "r") as file:
        new_dict = yaml.safe_load(file)

-    converted_dict = training_config.convert_legacy_training_config_dict(legacy_dict)
+    converted_dict = training_config.convert_legacy_training_config_dict(
+        legacy_dict
+    )

    for key, value in new_dict.items():
        assert converted_dict[key] == value