Merge remote-tracking branch 'fork/ml_fom' into ml_fom

cda-tum · Feb 18, 2025 · 0de9fd3 · 0de9fd3
2 parents f56c3a6 + 4deb521
commit 0de9fd3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mqt/predictor/ml/predictor.py b/src/mqt/predictor/ml/predictor.py
@@ -269,7 +269,7 @@ def generate_training_sample(
     def train_random_forest_model(
         self,
         save_model: bool = True,
-        device: Device = None,
+        device: Device | None = None,
     ) -> bool:
         """Trains a random forest model for the given figure of merit.