More typying

equinor · Oct 9, 2024 · 3b336a4 · 3b336a4
1 parent 2aae72e
commit 3b336a4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/ert/gui/tools/design_matrix/design_matrix_panel.py b/src/ert/gui/tools/design_matrix/design_matrix_panel.py
@@ -1,10 +1,14 @@
+from typing import Optional
+
 import pandas as pd
 from qtpy.QtGui import QStandardItem, QStandardItemModel
-from qtpy.QtWidgets import QDialog, QTableView, QVBoxLayout
+from qtpy.QtWidgets import QDialog, QTableView, QVBoxLayout, QWidget
 
 
 class DesignMatrixPanel(QDialog):
-    def __init__(self, design_matrix_df: pd.DataFrame, parent=None):
+    def __init__(
+        self, design_matrix_df: pd.DataFrame, parent=Optional[QWidget]
+    ) -> None:
         super().__init__(parent)
 
         self.setWindowTitle("Design matrix parameters viewer")