Skip to content

Commit

Permalink
fix: remove import reference of old package (#31)
Browse files Browse the repository at this point in the history
  • Loading branch information
isha97 authored Jan 31, 2025
1 parent 1bf3a09 commit 28651cb
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions google/cloud/spark_connect/session.py
Original file line number Diff line number Diff line change
Expand Up @@ -424,11 +424,11 @@ def _get_and_validate_version(self, dataproc_config, session_template):
]
import importlib.metadata

dataproc_connect_version = importlib.metadata.version(
"dataproc-spark-connect"
google_connect_version = importlib.metadata.version(
"google-spark-connect"
)
client_version = importlib.metadata.version("pyspark")
version_message = f"Spark Connect: {dataproc_connect_version} (PySpark: {client_version}) Session Runtime: {version} (Spark: {server_version})"
version_message = f"Spark Connect: {google_connect_version} (PySpark: {client_version}) Session Runtime: {version} (Spark: {server_version})"
logger.info(version_message)
if trimmed_version(client_version) != trimmed_version(
server_version
Expand Down

0 comments on commit 28651cb

Please sign in to comment.