#820

Committed 01 Oct 2023 02:22AM UTC coverage: 0.0% (-73.7%) from 73.748%

Build # #820

Build Type

push

circle-ci

Committed by Jiashen Cao

Commit Message

fix lint error

Run Details

0 of 12361 relevant lines covered (0.0%)

0.0 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/evadb/executor/project_executor.py

# coding=utf-8
# Copyright 2018-2023 EvaDB
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from typing import Iterator

import pandas as pd

from evadb.database import EvaDBDatabase
from evadb.executor.abstract_executor import AbstractExecutor
from evadb.executor.executor_utils import ExecutorError, apply_project
from evadb.models.storage.batch import Batch
from evadb.plan_nodes.project_plan import ProjectPlan


class ProjectExecutor(AbstractExecutor):
    """ """

    def __init__(self, db: EvaDBDatabase, node: ProjectPlan):
        super().__init__(db, node)
        self.target_list = node.target_list

    def exec(self, *args, **kwargs) -> Iterator[Batch]:
        # SELECT expr;
        if len(self.children) == 0:
            # Create a dummy batch with size 1
            dummy_batch = Batch(pd.DataFrame([0]))
            batch = apply_project(dummy_batch, self.target_list, self.catalog())
            if not batch.empty():
                yield batch
        # SELECT expr FROM table;
        elif len(self.children) == 1:
            child_executor = self.children[0]
            for batch in child_executor.exec(**kwargs):
                batch = apply_project(batch, self.target_list, self.catalog())
                if not batch.empty():
                    yield batch
        else:
            raise ExecutorError("ProjectExecutor has more than 1 children.")

1	# coding=utf-8
2	# Copyright 2018-2023 EvaDB
3	#
4	# Licensed under the Apache License, Version 2.0 (the "License");
5	# you may not use this file except in compliance with the License.
6	# You may obtain a copy of the License at
7	#
8	# http://www.apache.org/licenses/LICENSE-2.0
9	#
10	# Unless required by applicable law or agreed to in writing, software
11	# distributed under the License is distributed on an "AS IS" BASIS,
12	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13	# See the License for the specific language governing permissions and
14	# limitations under the License.
15	from typing import Iterator	×
16
17	import pandas as pd	×
18
19	from evadb.database import EvaDBDatabase	×
20	from evadb.executor.abstract_executor import AbstractExecutor	×
21	from evadb.executor.executor_utils import ExecutorError, apply_project	×
22	from evadb.models.storage.batch import Batch	×
23	from evadb.plan_nodes.project_plan import ProjectPlan	×
24
25
26	class ProjectExecutor(AbstractExecutor):	×
27	""" """
28
29	def __init__(self, db: EvaDBDatabase, node: ProjectPlan):	×
30	super().__init__(db, node)	×
31	self.target_list = node.target_list	×
32
33	def exec(self, args, *kwargs) -> Iterator[Batch]:	×
34	# SELECT expr;
35	if len(self.children) == 0:	×
36	# Create a dummy batch with size 1
37	dummy_batch = Batch(pd.DataFrame([0]))	×
38	batch = apply_project(dummy_batch, self.target_list, self.catalog())	×
39	if not batch.empty():	×
40	yield batch	×
41	# SELECT expr FROM table;
42	elif len(self.children) == 1:	×
43	child_executor = self.children[0]	×
44	for batch in child_executor.exec(**kwargs):	×
45	batch = apply_project(batch, self.target_list, self.catalog())	×
46	if not batch.empty():	×
47	yield batch	×
48	else:
49	raise ExecutorError("ProjectExecutor has more than 1 children.")

georgia-tech-db / eva / #820

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous