Skip to content

Commit

Permalink
[SPARK-44424][PYTHON][CONNECT][FOLLOW-UP] Import Connect related libr…
Browse files Browse the repository at this point in the history
…aries after checking dependencies

### What changes were proposed in this pull request?

This PR is a followup of #42235 that fixes the Connect related import after the dependency checking.

### Why are the changes needed?

In order to show the end users a nice error message for optional dependencies instead of just saying that they were not found.

### Does this PR introduce _any_ user-facing change?

No, the PR has not been released out yet.

### How was this patch tested?

Manually.

Closes #42311 from HyukjinKwon/SPARK-44424-folowup.

Authored-by: Hyukjin Kwon <gurwls223@apache.org>
Signed-off-by: Hyukjin Kwon <gurwls223@apache.org>
  • Loading branch information
HyukjinKwon committed Aug 2, 2023
1 parent 68d8e65 commit 7b9057b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/pyspark/sql/connect/client/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@
"getLogLevel",
]

from pyspark.sql.connect.client.reattach import ExecutePlanResponseReattachableIterator
from pyspark.sql.connect.utils import check_dependencies

check_dependencies(__name__)
Expand Down Expand Up @@ -66,6 +65,7 @@
from pyspark.version import __version__
from pyspark.resource.information import ResourceInformation
from pyspark.sql.connect.client.artifact import ArtifactManager
from pyspark.sql.connect.client.reattach import ExecutePlanResponseReattachableIterator
from pyspark.sql.connect.conversion import storage_level_to_proto, proto_to_storage_level
import pyspark.sql.connect.proto as pb2
import pyspark.sql.connect.proto.base_pb2_grpc as grpc_lib
Expand Down

0 comments on commit 7b9057b

Please sign in to comment.