diff --git a/src/ria_toolkit_oss/viz/pytorch_state_dict.py b/src/ria_toolkit_oss/viz/pytorch_state_dict.py
index 9bceb1c..578ebd0 100644
--- a/src/ria_toolkit_oss/viz/pytorch_state_dict.py
+++ b/src/ria_toolkit_oss/viz/pytorch_state_dict.py
@@ -1,6 +1,6 @@
+import numpy as np
 import plotly.graph_objects as go
 from plotly.graph_objects import Figure
-import numpy as np
 
 
 def create_styled_error_figure(title: str, message: str, suggestion: str = None) -> go.Figure: