Skip to content

Commit

Permalink
Currently when using throws an error because we are only sending the …
Browse files Browse the repository at this point in the history
…headers when calling (#669)
  • Loading branch information
ashishdatta authored Oct 22, 2023
1 parent 87dd966 commit afd52bc
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion pandasai/smart_dataframe/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -433,7 +433,10 @@ def _get_sample_head(self) -> DataFrameType:
sampler = DataSampler(head)
sampled_head = sampler.sample(rows_to_display)

return self._truncate_head_columns(sampled_head)
if self.lake.config.enforce_privacy:
return sampled_head
else:
return self._truncate_head_columns(sampled_head)

def _load_from_config(self, name: str):
"""
Expand Down

0 comments on commit afd52bc

Please sign in to comment.