addapt the script to the removed deprecated functions

weaviate · Feb 5, 2021 · a7969a9 · a7969a9
1 parent e7dbf57
commit a7969a9
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/integration/crud.py b/integration/crud.py
@@ -60,12 +60,12 @@ def _create_objects_batch(self):
         print("Create a batch with data.")
         objects_batch = weaviate.batch.ObjectsBatchRequest()
 
-        objects_batch.add_object({"name": "John Rawls"}, "Person")
-        objects_batch.add_object({"name": "Emanuel Kant"}, "Person")
-        objects_batch.add_object({"name": "Plato"}, "Person")
-        objects_batch.add_object({"name": "Pull-up"}, "Exercise")
-        objects_batch.add_object({"name": "Squat"}, "Exercise")
-        objects_batch.add_object({"name": "Star jump"}, "Exercise")
+        objects_batch.add({"name": "John Rawls"}, "Person")
+        objects_batch.add({"name": "Emanuel Kant"}, "Person")
+        objects_batch.add({"name": "Plato"}, "Person")
+        objects_batch.add({"name": "Pull-up"}, "Exercise")
+        objects_batch.add({"name": "Squat"}, "Exercise")
+        objects_batch.add({"name": "Star jump"}, "Exercise")
 
         print("Load batch.")
         self.client.batch.create(objects_batch)
@@ -98,10 +98,10 @@ def _create_references(self):
         reference_batch = weaviate.batch.ReferenceBatchRequest()
 
         for chemist in self.chemists:
-            reference_batch.add_reference("577887c1-4c6b-5594-aa62-f0c17883d9bf", "Group", "members",
+            reference_batch.add("577887c1-4c6b-5594-aa62-f0c17883d9bf", "Group", "members",
                                           chemist)
 
-        self.client.batch.add_references(reference_batch)
+        self.client.batch.create_references(reference_batch)
 
     def _validate_data_loading(self):
         print("Validate if loading was successful")