Fix leftover debugging code and fix linter errors

Signed-off-by: Jared O'Connell <[email protected]>

Author: jaredoconnell

tests/unit/benchmark/test_output.py

@@ -34,20 +34,10 @@ def test_generative_benchmark_marshalling():
     report = GenerativeBenchmarksReport(benchmarks=[mock_benchmark])
 
     serialized = report.model_dump()
-    print("Serialized: ", serialized)
     deserialized = GenerativeBenchmarksReport.model_validate(serialized)
-    print("Deserialized: ", deserialized)
     deserialized_benchmark = deserialized.benchmarks[0]
-    serialized = deserialized.model_dump()
-    print("Re-serialized: ", serialized)
 
-    print("Fields:", {name: getattr(deserialized, name) for name in deserialized.model_fields})
-    print("Extras:", deserialized.__pydantic_extra__)
-
-
-    #for field in mock_benchmark.model_fields_set:
-    #    assert getattr(mock_benchmark, field) == getattr(deserialized_benchmark, field)
-    # Workaround for duplicate fields for computed fields.
+    # model_dump as workaround for duplicate fields for computed fields.
     assert mock_benchmark.model_dump() == deserialized_benchmark.model_dump()
 
 
@@ -65,9 +55,7 @@ def test_file_json():
     loaded_report = GenerativeBenchmarksReport.load_file(mock_path)
     loaded_benchmark = loaded_report.benchmarks[0]
 
-    #for field in mock_benchmark.model_fields_set:
-    #    assert getattr(mock_benchmark, field) == getattr(loaded_benchmark, field)
-    # Workaround for duplicate fields for computed fields.
+    # model_dump as workaround for duplicate fields for computed fields.
     assert mock_benchmark.model_dump() == loaded_benchmark.model_dump()
 
     mock_path.unlink()
@@ -87,9 +75,7 @@ def test_file_yaml():
     loaded_report = GenerativeBenchmarksReport.load_file(mock_path)
     loaded_benchmark = loaded_report.benchmarks[0]
 
-    #for field in mock_benchmark.model_fields_set:
-    #    assert getattr(mock_benchmark, field) == getattr(loaded_benchmark, field)
-    # Workaround for duplicate fields for computed fields.
+    # model_dump as workaround for duplicate fields for computed fields.
     assert mock_benchmark.model_dump() == loaded_benchmark.model_dump()
 
     mock_path.unlink()