diff --git a/evadb/executor/load_executor.py b/evadb/executor/load_executor.py index 4a947da47c..c90c7ccec3 100644 --- a/evadb/executor/load_executor.py +++ b/evadb/executor/load_executor.py @@ -14,11 +14,12 @@ # limitations under the License. from evadb.database import EvaDBDatabase from evadb.executor.abstract_executor import AbstractExecutor +from evadb.executor.executor_utils import ExecutorError from evadb.executor.load_csv_executor import LoadCSVExecutor from evadb.executor.load_multimedia_executor import LoadMultimediaExecutor from evadb.parser.types import FileFormatType from evadb.plan_nodes.load_data_plan import LoadDataPlan -from evadb.executor.executor_utils import ExecutorError + class LoadDataExecutor(AbstractExecutor): def __init__(self, db: EvaDBDatabase, node: LoadDataPlan): @@ -30,8 +31,8 @@ def exec(self, *args, **kwargs): """ # invoke the appropriate executor - if self.node.file_options["file_format"] == None: - err_msg = f"Invalid file format, please use supported file formats: CSV | VIDEO | IMAGE | DOCUMENT | PDF" + if self.node.file_options["file_format"] is None: + err_msg = "Invalid file format, please use supported file formats: CSV | VIDEO | IMAGE | DOCUMENT | PDF" raise ExecutorError(err_msg) if self.node.file_options["file_format"] in [ FileFormatType.VIDEO,