added log check

pjwozny · Mar 3, 2023 · abc9374 · abc9374
1 parent 2ef9441
commit abc9374
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/scripts/evaluate_submission.py b/scripts/evaluate_submission.py
@@ -288,14 +288,16 @@ def compute_metrics(fetch_episode_states, trainer, framework, submission_file, e
 
         with open("episode_states.pkl", "wb") as f:
             pkl.dump(episode_states[0], f, protocol=pkl.HIGHEST_PROTOCOL)
+
+        if log_config and log_config["enabled"]:
 
-        #log mitigation rate counts of each country over time
-        wandb.log({"mitigation_rate Counts Across Time":construct_stacked_bar_chart(episode_states[0],
-                                    field="mitigation_rate_all_regions")})
+            #log mitigation rate counts of each country over time
+            wandb.log({"mitigation_rate Counts Across Time":construct_stacked_bar_chart(episode_states[0],
+                                        field="mitigation_rate_all_regions")})
 
-         #log mitigation rate counts of each country over time
-        wandb.log({"minimum_mitigation_rate Counts Across Time":construct_stacked_bar_chart(episode_states[0],
-                                    field="minimum_mitigation_rate_all_regions")})
+            #log mitigation rate counts of each country over time
+            wandb.log({"minimum_mitigation_rate Counts Across Time":construct_stacked_bar_chart(episode_states[0],
+                                        field="minimum_mitigation_rate_all_regions")})
 
         for feature in desired_outputs:
             feature_values = [None for _ in range(num_episodes)]