Nit fix in rag_experiment

nomadic-ml · Jan 9, 2025 · b039e5b · b039e5b
1 parent 6d6ffce
commit b039e5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nomadic/experiment/rag.py b/nomadic/experiment/rag.py
@@ -559,7 +559,7 @@ def run_inference_pipeline(params_dict: Dict[str, Any]) -> RunResult:
             "detailed_results": detailed_results,
             "mean_score": mean_score,
         }
-    if "enable_logging" in params_dict:
+    if params_dict.get("enable_logging"):
         # Save inference results to a file
         with open("inference_results.json", "w") as f:
             json.dump(inference_results, f)