dtype fix in Weakpredictor

Brown-SciML · Apr 2, 2024 · b290263 · b290263
1 parent b65e9a1
commit b290263
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ise/models/hybrid.py b/ise/models/hybrid.py
@@ -375,8 +375,8 @@ def __init__(
         super(WeakPredictor, self).__init__()
 
         # Initialize attributes
-        self.lstm_num_layers = lstm_num_layers
-        self.lstm_num_hidden = lstm_hidden_size
+        self.lstm_num_layers = int(lstm_num_layers)
+        self.lstm_num_hidden = int(lstm_hidden_size)
         self.input_size = input_size
         self.output_size = output_size
         self.ice_sheet = ice_sheet