chore: format

Author: anyangml

tests/metrics/conftest.py

@@ -84,7 +84,8 @@
         virial_mae=1.52447,
         virial_rmse_natoms=0.109107,
         virial_mae_natoms=0.0660086,
-    ),DirectPredictRecord(
+    ),
+    DirectPredictRecord(
         id=5,
         model_name="test_dp",
         task_name="AIMD_Chig",
@@ -244,7 +245,6 @@
         virial_rmse_natoms=None,
         virial_mae_natoms=None,
     ),
-    
 ]
 
 RECORDS_CALCULATOR = [

tests/metrics/test_post_process.py

@@ -28,7 +28,9 @@ def test_process_results_for_one_model(
             in caplog.text
         )
     assert result["generalizability_force_field_results"]["Weighted"] is None
-    assert result["generalizability_force_field_results"]["MoS2"]["energy_rmse"] == 500.8
+    assert (
+        result["generalizability_force_field_results"]["MoS2"]["energy_rmse"] == 500.8
+    )
 
     # Find differences between the calculator tasks and results
     calculator_task_differences = (

tests/metrics/test_visualization.py

@@ -15,7 +15,9 @@ def test_aggregate_ood_results_for_one_model(
     aggregator = ResultsFetcher()
     result = aggregator.aggregate_ood_results_for_one_model(model=model)
     np.testing.assert_almost_equal(result["Molecules"], desired=0.28470115, decimal=5)
-    np.testing.assert_almost_equal(result["Inorganic Materials"],desired=0.24101483, decimal=5)
+    np.testing.assert_almost_equal(
+        result["Inorganic Materials"], desired=0.24101483, decimal=5
+    )
     assert result["Catalysis"] is None
     with caplog.at_level(logging.WARNING):
         assert (