fix a small bug

microsoft · Nov 7, 2024 · 9e317cc · 9e317cc
1 parent 3b0d95f
commit 9e317cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rdagent/scenarios/kaggle/proposal/proposal.py b/rdagent/scenarios/kaggle/proposal/proposal.py
@@ -383,7 +383,7 @@ def convert_model_experiment(self, response: str, trace: Trace) -> KGModelExperi
         response_dict = json.loads(response)
         tasks = []
         model_type = response_dict.get("model_type", "Model type not provided")
-        if model_type not in KG_SELECT_MAPPING:
+        if not isinstance(model_type, str) or model_type not in KG_SELECT_MAPPING:
             raise ModelEmptyError(
                 f"Invalid model type '{model_type}'. Allowed model types are: {', '.join(KG_SELECT_MAPPING)}."
             )