fix LLM sample typo (#146)

hitsz-ids · Feb 27, 2024 · 54ee09b · 54ee09b
1 parent ad3d188
commit 54ee09b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdgx/models/LLM/single_table/gpt.py b/sdgx/models/LLM/single_table/gpt.py
@@ -472,7 +472,7 @@ def _sample_with_metadata(self, count, *args, **kwargs):
 
         # return result in pd.DataFrame
         final_columns = self.columns + self.off_table_features
-        return pd.DataFrame(self._result_list, columns=final_columns)
+        return pd.DataFrame(result, columns=final_columns)
 
     def _sample_with_data(self, count, *args, **kwargs):
         """
@@ -512,4 +512,4 @@ def _sample_with_data(self, count, *args, **kwargs):
 
         # return result
         final_columns = self.columns + self.off_table_features
-        return pd.DataFrame(self._result_list, columns=final_columns)
+        return pd.DataFrame(self.result, columns=final_columns)