Fixed style issues.

src/orqviz/loss_function.py

@@ -1,6 +1,7 @@
+import time
 from typing import Callable, Optional
+
 import numpy as np
-import time
 
 
 def _calculate_new_average(

src/orqviz/utils.py

@@ -27,4 +27,4 @@ def load_viz_object(filename: str) -> OrqVizObject:
         please use orqviz.io.load_viz_object""",
         DeprecationWarning,
     )
-    return _load_viz_object(filename)
+    return _load_viz_object(filename)

tests/orqviz/io_test.py

@@ -5,8 +5,8 @@
 
 from orqviz.elastic_band.data_structures import Chain
 from orqviz.hessians import get_Hessian
-from orqviz.scans import perform_1D_scan, perform_2D_scan
 from orqviz.io import OrqVizObject, load_viz_object, save_viz_object
+from orqviz.scans import perform_1D_scan, perform_2D_scan
 
 
 def SUM_OF_SINES(params):

tests/orqviz/loss_function_test.py

@@ -1,6 +1,7 @@
-from orqviz.loss_function import LossFunctionWrapper
 import numpy as np
 
+from orqviz.loss_function import LossFunctionWrapper
+
 
 def mock_function(params, b, c):
     return np.sum(params * b + c)

tests/orqviz/scans/scans_plots_test.py

@@ -2,7 +2,6 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
-import pytest
 
 from orqviz.io import load_viz_object, save_viz_object
 from orqviz.scans import perform_1D_scan, perform_2D_interpolation
@@ -12,7 +11,6 @@
     plot_2D_interpolation_result,
     plot_2D_scan_result,
 )
-from orqviz.io import load_viz_object, save_viz_object
 
 
 def SUM_OF_SINES(params):

tests/orqviz/scans/scans_test.py

@@ -11,13 +11,7 @@
     perform_2D_interpolation,
 )
 from orqviz.scans.data_structures import Scan1DResult, Scan2DResult
-from orqviz.scans.plots import (
-    plot_1D_scan_result,
-    plot_2D_interpolation_result,
-    plot_2D_scan_result,
-)
 from orqviz.scans.scans_2D import perform_2D_scan
-from orqviz.io import load_viz_object, save_viz_object
 
 
 def SUM_OF_SINES(params):

tests/orqviz/utils_test.py

@@ -45,4 +45,4 @@ def test_saving_and_loading_datatypes():
             loaded_data_object = load_viz_object("test")
         os.remove("test")
         assert isinstance(loaded_data_object, OrqVizObject.__args__)
-        assert type(loaded_data_object) == type(data_object)
+        assert type(loaded_data_object) == type(data_object)