fix numpy bug

ctlearn-project · Jun 4, 2024 · eb807f1 · eb807f1
1 parent 4501e2f
commit eb807f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ctlearn/run_model.py b/ctlearn/run_model.py
@@ -91,7 +91,7 @@ def run_model(config, mode="train", debug=False, log_to_file=False):
     if mode == "train":
         if "Training" not in config:
             config["Training"] = {}
-        validation_split = np.float(config["Training"].get("validation_split", 0.1))
+        validation_split = np.float64(config["Training"].get("validation_split", 0.1))
         if not 0.0 < validation_split < 1.0:
             raise ValueError(
                 "Invalid validation split: {}. "