fix: try002 for provider apache cassandra

apache · Apr 6, 2024 · d7123c7 · d7123c7
1 parent 6fd8f36
commit d7123c7
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/airflow/providers/apache/cassandra/hooks/cassandra.py b/airflow/providers/apache/cassandra/hooks/cassandra.py
@@ -154,7 +154,7 @@ def get_lb_policy(policy_name: str, policy_args: dict[str, Any]) -> Policy:
         if policy_name == "WhiteListRoundRobinPolicy":
             hosts = policy_args.get("hosts")
             if not hosts:
-                raise Exception("Hosts must be specified for WhiteListRoundRobinPolicy")
+                raise ValueError("Hosts must be specified for WhiteListRoundRobinPolicy")
             return WhiteListRoundRobinPolicy(hosts)
 
         if policy_name == "TokenAwarePolicy":

diff --git a/pyproject.toml b/pyproject.toml
@@ -388,8 +388,6 @@ combine-as-imports = true
 # amazon
 "airflow/providers/amazon/aws/operators/athena.py" = ["TRY002"]
 "airflow/providers/amazon/aws/operators/emr.py" = ["TRY002"]
-# apache.cassandra
-"airflow/providers/apache/cassandra/hooks/cassandra.py" = ["TRY002"]
 # cncf.kubernetes
 "airflow/providers/cncf/kubernetes/operators/pod.py" = ["TRY002"]
 # common.sql