This is an automated email from the ASF dual-hosted git repository.

husseinawala pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new 79cf7e0609 weaviate remove deprecations (#39707)
79cf7e0609 is described below

commit 79cf7e0609ef0640088ee80117e95046fa48063b
Author: Bowrna <[email protected]>
AuthorDate: Mon May 20 12:52:13 2024 +0530

    weaviate remove deprecations (#39707)
---
 tests/always/test_example_dags.py                            | 3 ---
 tests/system/providers/weaviate/example_weaviate_cohere.py   | 2 +-
 tests/system/providers/weaviate/example_weaviate_openai.py   | 2 +-
 tests/system/providers/weaviate/example_weaviate_operator.py | 8 ++++----
 4 files changed, 6 insertions(+), 9 deletions(-)

diff --git a/tests/always/test_example_dags.py 
b/tests/always/test_example_dags.py
index 79f11eb517..95083dad1f 100644
--- a/tests/always/test_example_dags.py
+++ b/tests/always/test_example_dags.py
@@ -63,9 +63,6 @@ IGNORE_AIRFLOW_PROVIDER_DEPRECATION_WARNING: tuple[str, ...] 
= (
     
"tests/system/providers/google/cloud/kubernetes_engine/example_kubernetes_engine_resource.py",
     
"tests/system/providers/google/cloud/life_sciences/example_life_sciences.py",
     "tests/system/providers/google/marketing_platform/example_analytics.py",
-    "tests/system/providers/weaviate/example_weaviate_cohere.py",
-    "tests/system/providers/weaviate/example_weaviate_openai.py",
-    "tests/system/providers/weaviate/example_weaviate_operator.py",
     # Deprecated Operators/Hooks, which replaced by common.sql Operators/Hooks
     "tests/system/providers/apache/drill/example_drill_dag.py",
     "tests/system/providers/jdbc/example_jdbc_queries.py",
diff --git a/tests/system/providers/weaviate/example_weaviate_cohere.py 
b/tests/system/providers/weaviate/example_weaviate_cohere.py
index 528e32e91e..f2a190bfcf 100644
--- a/tests/system/providers/weaviate/example_weaviate_cohere.py
+++ b/tests/system/providers/weaviate/example_weaviate_cohere.py
@@ -82,7 +82,7 @@ def example_weaviate_cohere():
         task_id="perform_ingestion",
         conn_id="weaviate_default",
         class_name="Weaviate_example_class",
-        input_json=update_vector_data_in_json["return_value"],
+        input_data=update_vector_data_in_json["return_value"],
     )
 
     embed_query = CohereEmbeddingOperator(
diff --git a/tests/system/providers/weaviate/example_weaviate_openai.py 
b/tests/system/providers/weaviate/example_weaviate_openai.py
index f3f4839e70..4030423681 100644
--- a/tests/system/providers/weaviate/example_weaviate_openai.py
+++ b/tests/system/providers/weaviate/example_weaviate_openai.py
@@ -80,7 +80,7 @@ def example_weaviate_openai():
         task_id="perform_ingestion",
         conn_id="weaviate_default",
         class_name="Weaviate_example_class",
-        input_json=update_vector_data_in_json["return_value"],
+        input_data=update_vector_data_in_json["return_value"],
     )
 
     embed_query = OpenAIEmbeddingOperator(
diff --git a/tests/system/providers/weaviate/example_weaviate_operator.py 
b/tests/system/providers/weaviate/example_weaviate_operator.py
index 36a6aa5412..ae4097006e 100644
--- a/tests/system/providers/weaviate/example_weaviate_operator.py
+++ b/tests/system/providers/weaviate/example_weaviate_operator.py
@@ -121,7 +121,7 @@ def example_weaviate_using_operator():
         task_id="batch_data_with_vectors_xcom_data",
         conn_id="weaviate_default",
         class_name="QuestionWithoutVectorizerUsingOperator",
-        input_json=store_data_with_vectors_in_xcom(),
+        input_data=store_data_with_vectors_in_xcom(),
         trigger_rule="all_done",
     )
     # [END howto_operator_weaviate_embedding_and_ingest_xcom_data_with_vectors]
@@ -131,7 +131,7 @@ def example_weaviate_using_operator():
         task_id="batch_data_with_vectors_callable_data",
         conn_id="weaviate_default",
         class_name="QuestionWithoutVectorizerUsingOperator",
-        input_json=get_data_with_vectors(),
+        input_data=get_data_with_vectors(),
         trigger_rule="all_done",
     )
     # [END 
howto_operator_weaviate_embedding_and_ingest_callable_data_with_vectors]
@@ -231,7 +231,7 @@ def example_weaviate_using_operator():
         task_id="batch_data_without_vectors_xcom_data",
         conn_id="weaviate_default",
         class_name="QuestionWithOpenAIVectorizerUsingOperator",
-        input_json=xcom_data_without_vectors["return_value"],
+        input_data=xcom_data_without_vectors["return_value"],
         trigger_rule="all_done",
     )
     # [END howto_operator_weaviate_ingest_xcom_data_without_vectors]
@@ -241,7 +241,7 @@ def example_weaviate_using_operator():
         task_id="batch_data_without_vectors_callable_data",
         conn_id="weaviate_default",
         class_name="QuestionWithOpenAIVectorizerUsingOperator",
-        input_json=get_data_without_vectors(),
+        input_data=get_data_without_vectors(),
         trigger_rule="all_done",
     )
     # [END howto_operator_weaviate_ingest_callable_data_without_vectors]

Reply via email to