#840

Committed 18 Oct 2023 09:51PM UTC coverage: 68.616% (-9.8%) from 78.391%

Build # #840

Build Type

push

circle-ci

Committed by

jiashenC

Commit Message

[BUMP]: v0.3.9+dev

Run Details

2 of 2 new or added lines in 1 file covered. (100.0%)

8634 of 12583 relevant lines covered (68.62%)

0.69 hits per line

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

47.06

/evadb/executor/sample_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

from evadb.database import EvaDBDatabase
from evadb.executor.abstract_executor import AbstractExecutor
from evadb.models.storage.batch import Batch
from evadb.plan_nodes.sample_plan import SamplePlan


class SampleExecutor(AbstractExecutor):
    """
    Samples uniformly from the rows.

    Arguments:
        node (AbstractPlan): The Sample Plan

    """

    def __init__(self, db: EvaDBDatabase, node: SamplePlan):
        super().__init__(db, node)
        self._sample_freq = node.sample_freq.value

    def exec(self, *args, **kwargs) -> Iterator[Batch]:
        child_executor = self.children[0]

        current = 0
        for batch in child_executor.exec():
            result_batch = batch[current :: self._sample_freq]
            result_batch.reset_index()
            yield result_batch
            current = (current - len(batch)) % self._sample_freq

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	1✔
16
17	from evadb.database import EvaDBDatabase	1✔
18	from evadb.executor.abstract_executor import AbstractExecutor	1✔
19	from evadb.models.storage.batch import Batch	1✔
20	from evadb.plan_nodes.sample_plan import SamplePlan	1✔
21
22
23	class SampleExecutor(AbstractExecutor):	1✔
24	"""
25	Samples uniformly from the rows.
26
27	Arguments:
28	node (AbstractPlan): The Sample Plan
29
30	"""
31
32	def __init__(self, db: EvaDBDatabase, node: SamplePlan):	1✔
33	super().__init__(db, node)	×
34	self._sample_freq = node.sample_freq.value	×
35
36	def exec(self, args, *kwargs) -> Iterator[Batch]:	1✔
37	child_executor = self.children[0]	×
38
39	current = 0	×
40	for batch in child_executor.exec():	×
41	result_batch = batch[current :: self._sample_freq]	×
42	result_batch.reset_index()	×
43	yield result_batch	×
44	current = (current - len(batch)) % self._sample_freq	×

georgia-tech-db / eva / #840

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