From 4ff32b91ea4a895c12fc24a70664e2592b753a9a Mon Sep 17 00:00:00 2001 From: Mark Date: Fri, 24 Feb 2023 14:24:49 -0800 Subject: [PATCH] Fixed minor bug --- celeste/plot.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/celeste/plot.py b/celeste/plot.py index 84dd7b3..a9314f4 100644 --- a/celeste/plot.py +++ b/celeste/plot.py @@ -8,16 +8,13 @@ from multiprocessing import Pool m = Path("model_data/current") -scaled = True - # Make "predicted reward" plots def plot_pred(src_model): plotting.predicted_reward( src_model, m / f"plots/predicted/{src_model.stem}.png", - device = torch.device("cpu"), - scaled = scaled + device = torch.device("cpu") ) # Make "best action" plots @@ -26,8 +23,7 @@ def plot_best(src_model): src_model, m / f"plots/best_action/{src_model.stem}.png", - device = torch.device("cpu"), - scaled = scaled + device = torch.device("cpu") ) # Make "actual reward" plots