This commit is contained in:
Xi Yan 2024-10-23 13:16:08 -07:00
parent 5e1323b5bf
commit d8bbce6f7c
2 changed files with 1 additions and 1 deletions

View file

@ -64,6 +64,7 @@ class PandasDataframeDataset(BaseDataset):
assert self.df is not None, "Dataset loading failed. Please check logs."
# note that we will drop any columns in dataset that are not in the schema
self.df = self.df[self.dataset_def.dataset_schema.keys()]
# check all columns in dataset schema are present

View file

@ -55,7 +55,6 @@ def data_url_from_file(file_path: str) -> str:
base64_content = base64.b64encode(file_content).decode("utf-8")
mime_type, _ = mimetypes.guess_type(file_path)
print(mime_type)
data_url = f"data:{mime_type};base64,{base64_content}"