Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
jiashenC committed Apr 5, 2023
1 parent dfe094f commit 304cb29
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 6 deletions.
4 changes: 3 additions & 1 deletion eva/optimizer/plan_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,9 @@ def _recursive_strip_exchange(plan: AbstractPlan, is_top: bool = False):
plan.append_child(child)

if isinstance(plan, ExchangePlan):
assert not is_top or len(plan.children) == 1, "Top ExchangePlan can only have 1 child."
assert (
not is_top or len(plan.children) == 1
), "Top ExchangePlan can only have 1 child."
return plan.children
else:
return [plan]
Expand Down
2 changes: 1 addition & 1 deletion test/integration_tests/test_like.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,11 @@
# See the License for the specific language governing permissions and
# limitations under the License.
import unittest
from test.util import shutdown_ray

from eva.catalog.catalog_manager import CatalogManager
from eva.configuration.constants import EVA_ROOT_DIR
from eva.server.command_handler import execute_query_fetch_all
from test.util import shutdown_ray


class LikeTest(unittest.TestCase):
Expand Down
8 changes: 4 additions & 4 deletions test/optimizer/test_cascade_optimizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,13 +19,13 @@
import pytest

from eva.catalog.catalog_manager import CatalogManager
from eva.experimental.ray.planner.exchange_plan import ExchangePlan
from eva.models.storage.batch import Batch
from eva.server.command_handler import execute_query_fetch_all
from eva.plan_nodes.project_plan import ProjectPlan
from eva.optimizer.plan_generator import PlanGenerator
from eva.plan_nodes.nested_loop_join_plan import NestedLoopJoinPlan
from eva.plan_nodes.project_plan import ProjectPlan
from eva.plan_nodes.storage_plan import StoragePlan
from eva.experimental.ray.planner.exchange_plan import ExchangePlan
from eva.optimizer.plan_generator import PlanGenerator
from eva.server.command_handler import execute_query_fetch_all


@pytest.mark.notparallel
Expand Down

0 comments on commit 304cb29

Please sign in to comment.